题目
编写一个算法来判断一个数 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
题解
题目从无限循环可以知道在寻找快乐数的时候,会重复访问数字。那么我们可以用一个set来记录访问的数字,若当前访问的数字不为1且不在set中,则将计算后的数放入set中。之后按照快乐数计算,若最后为1则return true。若不为1,并将当前查询数更换为计算后的数。重复如此,则最后退出循环后,一定return false。这时需要一个特判,若一开始n=1,按照之前逻辑则直接return false了。因此如果n=1,return
true。
bool isHappy(int n) {
set<int> records;
if(n==1)
return true;
while((n!=1)&&(records.find(n)==records.end())) {
records.insert(n);
string s = to_string(n);
int sum = 0;
for (char c:s) {
sum += (c - '0') * (c - '0');
}
if (sum != 1){
n=sum;
}
else return true;
}
return false;
}
Comments NOTHING