LeetCode 快乐数（Happy Number）

题目描述

 

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

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

示例: 

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

 

解题思路

 

用一个set记录之前出现过的数字，若遇到重复数字就停止循环。

 

代码

 

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