编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
示例 1:
输入:n = 19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2:
输入:n = 2
输出:false
思路:首先对于给出的数,需要求出它的每一位的数字才能求出平方和sum,因此可以利用循环除10操作,再求余以得到每次的个位数,例:
int n=789;
int a=0;
int b=0;
int c=0;
a =n%10; //a=9
n/=10; //m=78
b=n%10; //b=8
n/=10; //n=7
c=n%10; //c=7
//a=9,b=8,c=7
方法一:
再看回来题目,题目说会出现无限循环,即如果sum重复出现的时候就会陷入循环中,即只需要判断在循环中sum是否已经出现过即可判断该循环是否会陷入无限循环,这里可以使用c++的unordered_set,把每次求出的sum放入集合,在放入之前再做一次判断sum是否已存在,若存在即直接返回false即可。
class Solution {
public:
int getSum(int n){
int sum=0;
while(n){
sum+=(n%10)*(n%10);
n/=10;
}
return sum;
}
bool isHappy(int n) {
unordered_set<int> set;
int sum=0;
while(true){
sum=getSum(n);
if(sum==1){
return true;
}
if(set.find(sum)!=set.end()){
return false;
}
else{
set.insert(sum);
}
n=sum;
}
}
};
方法二:
第二种方法就是可以使用快慢指针法来跳出循环,慢指针每次走一步,快指针每次走两步,如果sum出现了循环,则慢指针和快指针最终肯定会相遇(相等),此时即可跳出循环,最后只需判断跳出循环的条件是否是因为slow==1即可
class Solution {
public:
int getSum(int n){
int sum=0;
while(n){
sum+=(n%10)*(n%10);
n/=10;
}
return sum;
}
bool isHappy(int n) {
int slow=n;
int fast=n;
do{
slow=getSum(slow);
fast=getSum(fast);
fast=getSum(fast);
} while(fast!=slow);
return slow==1;
}
};