LeetCode788. 旋转数字

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

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

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

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

class Solution(object):
    def rotatedDigits(self, N):
        """
        :type N: int
        :rtype: int
        """
        # 每位都在(2, 5, 6, 9, 0, 1, 8)内，至少一位在(2, 5, 6, 9)内
        change = ['2', '5', '6', '9']
        able = ['2', '5', '6', '9', '0', '1', '8']
        count = 0
        for i in range(1,N+1):
            # able_flag和change_flag分别记录是否有出现在change和able中的数字
            able_flag = True
            change_flag = False
            
            for c in str(i):
                # 如果有数字在change中，就为True，因为我们规定只要有一个数字在change中就可以
                if c in change:
                    change_flag = True
                # 如果有数字不在able中，就为False，因为我们规定每个数字都必须在able中
                if c not in able:
                    able_flag = False
            # 两个条件都满足的时候count+=1
            if able_flag and change_flag:
                count+=1
        return count