题目描述:
编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」 定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
示例 1:
输入:n = 19
输出:true
解释:
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1
示例 2:
输入:n = 2
输出:false
解法一:Set
思路:
在每次求和过程中,sum可能会重复出现。如果出现重复则代表陷入死循环。因此可以使用set/HashMap/List等判断这个sum有没有出现。如果已经出现过,则直接返回false;否则就一直循环直到sum为1停止。
代码:
class Solution {
public boolean isHappy(int n) {
//存储sum
Set<Integer> set = new HashSet<>();
//循环
while(n != 1 && !set.contains(n)){
set.add(n);
n = getNext(n);
}
//判断n是否等于1
return n == 1;
}
public int getNext(int n){
int sum = 0;
while(n > 0){
//拆分
int val = n % 10;
n /= 10;
//计算
sum += val * val;
}
return sum;
}
}
解法二:快慢指针
思路:
使用集合记录每次的计算结果,会出现集合很大的情况。因此可以使用快慢指针的思想,快指针每次走两步,慢指针每次走一步。当快慢指针相等的时候,判断循环的引起是否是1引起的循环,如果是则为快乐数,反之则不是。
代码:
class Solution {
public boolean isHappy(int n) {
int slow = n, fast = getNext(n);
while(slow != fast){
slow = getNext(slow);
fast = getNext(getNext(fast));
}
return slow == 1;
}
public int getNext(int n){
int sum = 0;
while(n > 0){
//拆分
int val = n % 10;
n /= 10;
//计算
sum += val * val;
}
return sum;
}
}