编写一个算法来判断一个数是不是“快乐数”。
一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
示例:
输入: 19
输出: true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
解:计算出下一个数,如果出现重复则不可能为1,会无限循环,返回false,如果为1则返回true。关键是getNext求下一个数。
public boolean isHappy(int n) {
int temp = n;
HashSet<Integer> hashSet = new HashSet<Integer>();
hashSet.add(temp);
while (true) {
temp = getNext(temp);
if (temp == 1) {
return true;
} else if (hashSet.contains(temp)) {
return false;
}
hashSet.add(temp);
}
}
private int getNext(int num) {
int result = 0;
while (num > 0) {
int a = (num % 10);
int b = (num % 10);
result += a * b;
num = num / 10;
}
return result;
}