题目
编写一个算法来判断一个数 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
代码
关键理解:
- 无限循环意味着会出现和相同的情况
- 熟练掌握求整数上每个位
class Solution {
public:
bool isHappy(int n) {
unordered_set<int> result;
while (1) {
int sum = getSum(n);
if (sum == 1) {
return true;
}
else if (result.find(sum) != result.end()) {
// 找到相同数字
return false;
}
else {
result.insert(sum);
}
n = sum; // 将上一轮求和得到的数字作为新的待判定正整数
}
}
int getSum(int n) {
int sum = 0;
while (n) {
// 求数字上每个位的平方和
sum += (n % 10) * (n % 10);
n /= 10;
}
return sum;
}
};