编写一个算法来判断一个数 n
是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n
是 快乐数 就返回 true
;不是,则返回 false
。
-
解题思路
题目中提到了“无限循环”,因此在重复这个过程的途中,一定会遇到重复的数,否则不会无限循环!这样我们就可以设置一个Set,判断sum是否重复出现。只要sum为1,就可以返回ture;只要sum重复出现,则返回false。 -
代码
class Solution {
public boolean isHappy(int n) {
Set<Integer> set = new HashSet<>();
while (true) {
int sum = isHappyHelper(n);
if (sum == 1) {
return true;
}
if (set.contains(sum) ){
return false;
} else {
set.add(isHappyHelper(n));
}
n = sum;
}
}
public int isHappyHelper(int num){
int sum = 0;
while (num != 0){
sum += (num % 10) * (num % 10);
num = num / 10;
}
return sum;
}
}