LeetCode 202. 快乐数(C++)

题目：

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

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

示例:

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

思路：

对于快乐数，难度在于如何每个位置上的数字的平方和，这里我们可以对n取余数得到n的个位，求其平方并累加，对n /10的结果用int保存，得到去除个位的n的其他位数字，重复上述步骤，直至其他位数字求取结果为零，得到n的各个位置的数字的平方和，重复操作直至平方和等于1。对于非快乐数，肯定不能进入死循环，那么肯定循环进行到一定程度，数字会和之前出现过n重复，因此我们保存每一步骤的n，并在得到新的n的时候判断该数字是否出现过，出现过就为非快乐数，跳出循环。

class Solution {
public:
    bool isHappy(int n) {
        set<int> recordN;        
        int sum = 0;
        while(n != 1){
            recordN.insert(n);
            sum = 0;
            while(n != 0){
                sum += ((n % 10) * (n % 10));
                n = n / 10;
            }
            if(recordN.find(sum) != recordN.end())
                return false;
            n = sum;
        }
        return true;
    }
};