题目
思路
要注意题目中说的无限循环:它是指在求平方和的过程中,会再次出现之前的值(想象一个圈),这种情况的时候肯定算不出1来。
所以我们要设定跳出循环的条件是:当平方和结果为1或者出现循环了
出现循环怎么用代码表示呢?我们可以用一个HashSet记录求过的平方和,每次新求出一个平方和来,都看看HashSet里有没有这个值,有的话就说明出现了重复。
import java.util.HashSet;
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public boolean isHappy(int n) {
Set<Integer> record = new HashSet<>();
//当结果为1 或者 无限循环的时候跳出这个循环
while (n != 1 && !record.contains(n)) {
record.add(n);
//获得每位数字平方和
n = getSum(n);
}
return n == 1;
}
public int getSum(int n) {
int res = 0;
while (n > 0) {
int temp = n % 10;
res = res + temp * temp;
n = n / 10;
}
return res;
}
}
//leetcode submit region end(Prohibit modification and deletion)