leetcode202:快乐数

思想:

定义字典record记录正整数每个位置上的数字的平方和,每次计算出来的平方和都保存。若平方和出现重复则不是快乐数,反之则是。变量sq_num记录每次的平方和。将n先求余10得到个位上的数并平方,然后整除10得到去除个位上数的其他数,n其次进行循环直到结束计算出平方和,然后判断平方和有没有出现过,若没有则继续算,反之则返回false。

class Solution:
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        record = {}
        sq_sum = 0
        while n != 1:
            sq_sum = 0
            sub_num = n
            while sub_num > 0:
                sq_sum += (sub_num % 10) * (sub_num % 10)
                sub_num //= 10
            if sq_sum in record:
                return False
            else:
                record[sq_sum] = 1
            n = sq_sum
        return True

都是大佬们的思想,小菜鸟还是不行的

猜你喜欢

转载自blog.csdn.net/weixin_43160613/article/details/83714708