判断快乐数-哈希表

快乐数-力扣

思路

• 快乐数：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是无限循环但始终变不到 1。如果可以变为1，就是快乐数；否则，非快乐数。
• 破题点：无限循环，也就是平方和会重复出现，则可看做判断一个数是否在一个集合里，考虑用哈希法
• 解决办法：判断sum值是否重复出现，若是，return False；否则，一直找到sum为1才停止；可用python中的set()。
• 难点：取数值各个位上的单数 (n%10)**2为平方和，n//10为取单数

class Solution:
    def isHappy(self, n: int) -> bool:
        #记录中间结果
        tmp_set=set()
        while True:
            sum=self.getSum(n)
            #出现1,是快乐数
            if sum == 1:
                return True
            #重复出现，说明已出现循环，肯定不是快乐数
            if sum in tmp_set:
                return False
            else:
                tmp_set.add(sum)
            n=sum


    def getSum(self,n:int)->int:
        sum=0
        while n:
            #求平方和
            sum+=(n%10)**2
            #商取整，得单数
            n=n//10
        return sum