编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」 定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。 如果这个过程 结果为 1,那么这个数就是快乐数。 如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
由于这个过程可能无限循环,即不满足情况时多个数字无限循环 所以可以使用Set来将每次平方后的数字记录 ,若发现有重复则返回false;
则代码如下:
class Solution {
public boolean isHappy(int n) {
Set<Integer> set = new HashSet<>();
while(n !=1 && !set.contains(n) ){
set.add(n);
n = getNextNumber(n);
}
return n == 1;
}
private int getNextNumber(int n) {
int ret = 0;
while(n > 0){
int temp = n %10;
ret += temp*temp;
n /= 10;
}
return ret;
}
}