旋转数字

63 阅读1分钟

788. 旋转数字 - 力扣(LeetCode)

我们称一个数 X 为好数, 如果它的每位数字逐个地被旋转 180 度后,我们仍可以得到一个有效的,且和 X 不同的数。要求每位数字都要被旋转。

如果一个数的每位数字被旋转以后仍然还是一个数字, 则这个数是有效的。0, 1, 和 8 被旋转后仍然是它们自己;2 和 5 可以互相旋转成对方(在这种情况下,它们以不同的方向旋转,换句话说,2 和 5 互为镜像);6 和 9 同理,除了这些以外其他的数字旋转以后都不再是有效的数字。

现在我们有一个正整数 N, 计算从 1 到 N 中有多少个数 X 是好数?

示例:

输入: 10
输出: 4
解释:
在[1, 10]中有四个好数: 2, 5, 6, 9。
注意 1 和 10 不是好数, 因为他们在旋转之后不变。

提示:

  • N 的取值范围是 [1, 10000]

解题

/**
 * @param {number} n
 * @return {number}
 */
var rotatedDigits = function (n) {
  const idxs = [0, 1, 2, 3, 3, 3, 4, 5, 5, 6];
  const nums = Array.from(n.toString()).map((v) => +v);
  const len = nums.length;
  const getCount = (num, bit, removeSame) => {
    if (num === 0) return 0;
    let count = idxs[num];
    let sameCount = !removeSame ? 0 : num > 8 ? 3 : num > 1 ? 2 : 1;
    for (let i = 0; i < bit - 1; i++) {
      count *= 7;
      sameCount *= 3;
    }
    return count - sameCount;
  };
  let res = 0;
  let removeSame = true;
  let i= 0
  while(i<len){
    res +=getCount(nums[i], len-i, removeSame)
    if([3,4,7].includes(nums[i])){
      break
    }else{
      removeSame = removeSame && [0, 1, 8].includes(nums[i]);
      i++
    }
  }
  if (i===len && !removeSame) {
    res += 1;
  }
  return res;
};