LeetCode 788. 旋转数字（python）

题目链接

题目描述：

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

如果一个数的每位数字被旋转以后仍然还是一个数字， 则这个数是有效的。0, 1, 和 8 被旋转后仍然是它们自己；2 和 5 可以互相旋转成对方；6 和 9 同理，除了这些以外其他的数字旋转以后都不再是有效的数字。

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

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

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

解题思路：

对于1到N的每一个数，对10取余，判断余数是否是可旋转的，如果不能直接判断下一个数，如果能，判断一下是否旋转完是否是不同的数字（2，5，6，9），是即将flag记为1，然后将该数字整除10，重复上述操作，直到该数字为0为止。

class Solution:
    def rotatedDigits(self, N: int) -> int:
        res1=[0,1,8]
        res2=[2,5,6,9]
        count=0
        for i in range(1,N+1):
            flag=0
            while i!=0:
                x=i%10
                if x in res2:
                    flag=1#旋转成不同的数
                if x not in res1 and x not in res2:#不能被旋转
                    flag=0
                    break
                i=i//10
            if flag==1:
                count+=1
        return count