编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 true ;不是,则返回 false 。
示例 1:
输入:19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2:
输入:n = 2
输出:false
提示:
1 <= n <= 231 - 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/happy-number
方法一: 数学方式
首先我们依据题目,表明最后的平方和会变成一个小于10的数,此时我们只要判断这个小于10的数是否是一个平方数即可。基于此,这个将作为结束递归的条件。
对应的代码:
class Solution {
/*
数学方法解决
*/
public boolean isHappy(int n) {
if(n == 1 || n == 7) //上面的推理中,如果最后n是等于1或者7的话直接返回true,否则小于10的其他数字返回false
return true;
else if(n < 10)
return false;
//如果当前的数字大于等于10,那么就计算每个位置的数字的平方和,然后进入递归
int sum = 0;
while(n != 0){
sum += (int)Math.pow(n % 10, 2);
n /= 10;
}
return isHappy(sum);
}
}
运行结果:
方法二:利用哈希表判断是否进入了循环
class Solution {
public boolean isHappy(int n){
HashMap<Integer,Boolean> map = new HashMap<Integer,Boolean>();
int sum;
/*
获取当前n每一个位置的数字平方和,判断是否等于1,如果是,说明是平方和,如果不是,那么就将其
添加到哈希表中,然后计算sum的每一个数字的平方和,更新sum,重复这个操作
*/
sum = getSum(n);
while(sum != 1 && !map.containsKey(sum)){
map.put(sum,true);
sum = getSum(sum);
}
return sum == 1;
}
//获取n的每个位置上的数字的平方和
public int getSum(int n){
int sum = 0;
while(n != 0){
sum += (int)Math.pow(n % 10,2);
n /= 10;
}
return sum;
}
}
运行结果:
方法三:快慢指针
由于上面的推理中,可以得出有可能进入无限循环的情况,所以这时候利用快慢指针,fast指针每次移动2步,slow每次移动1步,那么如果没有无线循环,表示这个数字是一个快乐数,否则不是。
所以对应的代码:
class Solution {
public boolean isHappy(int n){
int slow,fast,tmp;
slow = fast = n;
do{
slow = getSum(n);
tmp = getSum(fast);//这里计算fast的下一步
fast = getSum(tmp);//计算tmp的下一步作为fast
n = slow;
}while(n != 1 && slow != fast);
return n == 1;//如果退出循环的时候,n等于1,表明是一个快乐数,否则快指针和慢指针相等,进入了循环,此时直接返回false
}
//获取n的每个位置上的数字的平方和
public int getSum(int n){
int sum = 0;
while(n != 0){
sum += (int)Math.pow(n % 10,2);
n /= 10;
}
return sum;
}
}
运行结果: