788_rotated_digits

788. 旋转数字 简单

我们称一个数 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]。

代码参考;

package main

import "fmt"

func main() {
    fmt.Println(rotatedDigits(2))
}

// 审好题，有逻辑就 ok
func rotatedDigits(N int) int {
    counts := 0
    for n := 1; n <= N; n++ {
        if valid, res := rotate(n); valid && n != res {
            counts++
        }
    }
    return counts
}

func rotate(num int) (valid bool, res int) {
    valid = true
    var nums []int
    for num > 0 {
        nums = append(nums, num%10) // 分割
        num /= 10
    }

    for i, n := range nums { // 旋转
        switch n {
        case 0, 1, 8:
        case 2:
            nums[i] = 5
        case 5:
            nums[i] = 2
        case 6:
            nums[i] = 9
        case 9:
            nums[i] = 6
        default:
            valid = false
        }
    }

    fmt.Println(nums)
    if valid {
        for i, n := range nums { // 重组
            res += n * pow(10, i)
        }
    }
    return
}

func pow(x, y int) int {
    res := 1
    for y > 0 {
        res *= x
        y--
    }
    return res
}