LeetCode-202. 快乐数

202. 快乐数


编写一个算法来判断一个数是不是“快乐数”。

一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。

示例: 

输入: 19
输出: true
解释: 
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

解题思路:使用集合s存储已经计算的数字,只要当前数字不在集合中,就替换其为它每个位置上的数字的平方和,直到该数字为1或者该数字出现在集合中。

Python3代码如下:

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        s = set()
        while n not in s:
            s.add(n)
            temp = 0
            for c in map(int,str(n)):
                temp += c*c
            n = temp
            if n == 1:
                return True
        return False

猜你喜欢

转载自blog.csdn.net/qq_36309480/article/details/89683000