题目描述:
编写一个算法来判断一个数是不是“快乐数”。
一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 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;
}
};