编写一个算法来判断一个数是不是“快乐数”。
一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
输入:19
输出:true
解释:
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1
解法
根据描述,整数N每一位平方求和结果是转换到另一个N,需要判断最终是否是所有位数的平方和为1,如果平方和为1,在执行每一位求和平均的操作,那么计算结果一直为1,题目已经提醒会出现无限循环且结果不为1的情况,因此此类问题可以看做一个链表找环的问题,解决方法如下:
public boolean isHappy(int n) {
Set<Integer> set = new HashSet<>();
while (true) {
int sum = 0;
while (n > 0) {
sum += (n % 10) * (n % 10);
n /= 10;
}
if (sum == 1) {
return true;
} else if (set.contains(sum)) {
return false;
} else {
set.add(sum);
n = sum;
}
}
}