力扣刷题-哈希表快乐数

202 快乐数

编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为 1，那么这个数就是快乐数。
如果 n 是快乐数就返回 True ；不是，则返回 False 。
示例：
输入：19
输出：true
解释：
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

思路

题目中说了会 无限循环，那么也就是说求和的过程中，sum会重复出现，这对解题很重要！
当我们遇到了要快速判断一个元素是否出现集合里的时候，就要考虑哈希法了。——>集合或数组
所以这道题目使用哈希法，来判断这个sum是否重复出现，如果重复了就是return false， 否则一直找到sum为1为止。（可以用集合，直接判断元素是否已经在集合中：if n in set；或者使用数组，循环条件为元素不在数组中：while n not in result）

使用集合

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        result = set() 
        while True: # 无限循环
            n = self.get_sum(n) # 注意这里必须赋值给 n 因为是无限循环
            if n == 1:
                return True # 结果为1 那就是快乐数
            
             # 如果中间结果重复出现，说明陷入死循环了，该数不是快乐数
            if n in result:
                return False
            
            else:
                result.add(n)



    # 首先要知道怎么计算一个数上每位数的平方和
    # 有两种方法
    # 法一 采用取模运算的方式来求每位数
    # def get_sum(self, n): # self
    #     sum = 0
    #     while n: # 当n不是0
    #         n, r = divmod(n, 10) # 第一次n为商 r为余数(个位数的数)
    #         sum += r ** 2
    #     return sum # 求得平方和

    # 法二 采用转换为字符串 求每位相加平方和的方式
    def get_sum(self, n):
        sum = 0
        str_n = str(n)
        for i in str_n:
            sum += int(i) ** 2
        return sum

使用集合+精简

class Soulution:
    def isHappy(self, n):
        result = set()
        while n != 1:
            # sum = 0
            # for i in str(n):
            #     sum += int(i)**2
            # n = sum
            n = sum(int(i)**2 for i in str(n))
            if n in result:
                return False
            result.add(n)
        return True

使用数组

class Solution:
   def isHappy(self, n: int) -> bool:
       record = []
       while n not in record: # 判断条件
           record.append(n)
           new_num = 0
           n_str = str(n)
           for i in n_str:
               new_num+=int(i)**2
           if new_num==1: return True
           else: n = new_num
       return False