leetcode202. 快乐数

题目描述

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

「快乐数」 定义为：

对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为 1，那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

示例 1：

输入：n = 19
输出：true

解释：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2：

输入：n = 2
输出：false

提示：

1 <= n <= 231 - 1

分析思路

这道题首先想到的是一个函数，用来求一个数各个位数的平方和。然后在主函数中用一个while(1)循环来重复判断这个和是否等于1。

注意：题目中说无限循环是什么意思呢？就是说，如果这个数不是快乐数，那么在set集合中，一定会出现sum值循环，而且会多次出现。借用评论区的一个回答：
为啥一定不会出现死循环，因为int类型最大值为为‭‭2 147 483 647‬‬， 所以平方和最大的数是1 999 999 999，平方和为1 + 81*9 = 724。任何数的平方和都在1到724之间，724次循环之内一定有重复的。

需要用一个unordered_set来存储sum和，具体代码如下：

class Solution {
public:

    int getSum(int n){
        int sum = 0;
        while(n){
            sum += (n%10) * (n%10);
            n /= 10;
        }
        return sum;
    }

    bool isHappy(int n) {
        
        unordered_set<int> res;
        // 无限循环
        while(1){
            int sum = getSum(n);
			// 如果等于1，返回true
            if(sum == 1){
                return true;
            }
			// 如果集合中找到了这个值，返回false
            if(res.find(sum)!=res.end()){
                return false;
            }else{
            // 否则插入
                res.insert(sum);
            }
            // 给n赋一个新的值
            n = sum;
        }
    }
};