编写一个算法来判断一个数 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
思路
1. 这题咋一看像是个数学题,而且看到循环两个字,大家一定会想到用递归,但是这是无限循环,潜在意思其实是会出现相同的数,既然会出现相同的数再往后进行判断就没有必要了。这个思路这对解题很重要。
2. 当我们遇到了要快速判断一个元素是否重复出现集合里的时候,需要我们快速想到哈希,这道题使用hash来判断sum是否重复出现吗,如果出现了直接返回false就好了,否则就一直循环下去到sum==1 为止
3. 首先需要写一个函数用来求n各个位置上的数的平方和,我们可以通过反复取余求每个位置上的数,并且一次次除10,去除掉已经求过平方的数,最后相加即可
代码如下:
class Solution {
public:
// 取数值各个位上的单数平方之和
int getSum(int n){
int sum = 0;
while (n){
sum += (n%10) * (n%10);
n = n/10;
}
return sum;
}
bool isHappy(int n) {
if(sum == 1)
return true;
int sum = getSum(n);
unordered_map<int,int> s;
while (sum != 1){
s[sum]++;
// 如果这个sum曾经出现过,说明已经陷入了无限循环了,立刻return false
if (s[sum] > 1)
return false;
sum = getSum(sum);
}
return true;
}
};