LeetCode 202.Happy Number (快乐数)

题目描述：

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

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

示例: 

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

AC C++ Solution:
 

class Solution {
public:
    bool isHappy(int n) {
        
        if(n == 0) return 0;
        
        set<int> s;         //用一个set存储每次平方之后的数字
        int sum = 0;
        while(1)
        {
            int tmp = n;
            sum = 0;
            while(tmp > 0)
            {
                int a = tmp%10;
                sum += a*a;
                tmp = tmp /10;
            }
            auto itr = s.find(sum);
            if(itr != s.end()) break;       //当在set中查找到相同的数字时，退出循环
            s.insert(sum);
            n =sum;          //n设置为本次sum的值,作为下一次循环的初始值
        }
        if(sum ==1) return 1;
        return 0;
        
    }
};