Python Leetcode(788. 旋转数字)

Python Leetcode(788. 旋转数字)

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

思路:好数包括2，5，6，9，坏数包括3，4，7，需要搭配形成好数的数字包括1，8，0.对于小于10的数字直接判断是否是好数，大于10的数字判断每一位数字是否属于坏数，且所有位的数字不同时为1，8，0

class Solution(object):
    def rotatedDigits(self, N):
        """
        :type N: int
        :rtype: int
        """
        count = 0
        
        good_num = [2, 5, 6, 9]
        bad_num = [3, 4, 7]
        medium_num = [0, 1, 8]
        
        for i in range(1, N+1):
            if i <= 10:
                if i in good_num:
                    count += 1
            else:
                flag = 0 # 记录数字为medium_num的个数
                position = 0 # 记录位数，如果position==flag，则这个数为坏数
                while i != 0:
                    position += 1
                    judge_num = i % 10
                    if judge_num in medium_num:
                        flag += 1
                    if judge_num not in bad_num:
                        i = i // 10
                        continue
                    break
                if i == 0 and flag != position:
                    count += 1
        return count

solution = Solution()
solution.rotatedDigits(857)

247