202. 快乐数

题目链接:https://leetcode-cn.com/problems/happy-number/

description：

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

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

示例:

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

思路：

这道题主要难点是什么时候输出false，我们可以借用一个set来将中间结果保存起来，如果在判断的过程中，发现又出现了这个数，那么说明肯定不可以得到1，否则就可以。

python3 代码实现：

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """


        def f(arr):
            sum = 0
            while len(arr) > 0:
                temp = arr.pop()
                sum += temp ** 2
            if sum in set_num:
                return False
            else:
                set_num.add(sum)
            if sum != 1:
                for k in str(sum):
                    arr.append(int(k))
                return f(arr)
            else:
                return True

        def check(s):
            arr = []
            if s == '1':
                return True
            for i in s:
                arr.append(int(i))
            return f(arr)

        set_num = set()
        return check(str(n))



s = Solution()
n = int(input())
rs = s.isHappy(n)
print(rs)

JavaScript实现：

/**
 * @param {number} n
 * @return {boolean}
 */
var isHappy = function(n) {
    n+=''
    sum = 0
    sum+=n
    set = new Set()
    while (sum != 1){
      sum = 0
        for (let i=0;i<n.length;i++){
            sum += n[i]*n[i];
        }
        if (set.has(sum)) {
            return false;
        }
        set.add(sum)
        n = ''+sum
    }
    return true
        
};