编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 True ;不是,则返回 False 。
示例:
输入:19
输出:true
解释:
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1
题目中说了会 无限循环,那么也就是说求和的过程中,sum会重复出现,这对解题很重要!
当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。
所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。
判断sum是否重复出现就可以使用unordered_set。
还有一个难点就是求和的过程,如果对取数值各个位上的单数操作不熟悉的话,做这道题也会比较艰难。
class Solution {
public:
// 取数值各个位上的单数之和
int getSum(int n)
{
int sum = 0;
while (n) {
sum += (n % 10) * (n % 10); //n%10是对10取余,就可以得到个位上的单数
n /= 10; //除以10,把最后一位去掉,因为上一步已经计算完了。
}
return sum;
}
bool isHappy(int n)
{
unordered_set<int> set; //定义set集合
while(1)
{
int sum = getSum(n); 得到sum
if (sum == 1)
{
return true;
}
// 如果这个sum曾经出现过,说明已经陷入了无限循环了,立刻return false
if (set.find(sum) != set.end()) //如果set里面有sum则返回sum的迭代器,return false
{
return false;
}
else
{
set.insert(sum); //set里面没有sum,则添加sum。
}
n = sum;
}
}
};
注意:
1.根据n,如何求sum
2.注意题意,要么是1,要么无限循环,就这两种结果,如果是1,直接return true。如果是无限循环,说明sum值会重复出现,如果判断sum是重复出现的?那就是第一次在set里面添加了sum后,再判断set中是否有sum。(判断set中是否有某个元素,就是哈希表的经典用法)。
参考:代码随想录