LeetCode 202. 快乐数

题目描述： 快乐数

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

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

案例: 19 是一个快乐数。

• 12 + 92 = 82
• 82 + 22 = 68
• 62 + 82 = 100
• 12 + 02 + 02 = 1

解题思路：

        判断出现的数之前有没有出现过，出现过就会产生循环，就不是快乐数。

代码：

class Solution {
public:
    bool isHappy(int n) {
        int ans = n;
        set<int> s;
        while(1) {
            if(ans == 1) return true;
            if(s.count(ans) == 1) return false;
            s.insert(ans);
            ans = is1(ans);
        }
    }
    
    int is1(int n) {
        int re = 0;
        while(n!= 0) {
            re += (n%10)*(n%10);
            n /= 10;
        }
        return re;
    }
};