题目(leecode T202):
编写一个算法来判断一个数 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
方法:分析题目的条件发现,我们要对一个数重复做算每个数字的平方和的操作,这部可以单独用一个函数getSum来运算。然后题目说有可能无限循环但变不到1,就意味着我们需要保存下来之前的所有平方和的结果。并判断新的运算结果有没有在以前出现过并且不为1,这样的循环就代表着永远也不会变成1了。而上一篇的unordered_set数据结构正好适用于快速查找一个新的元素之前是否出现过的情况,因此我们可以使用unordered_set,每次将getSum的结果保存到其中,出现1就返回true,出现循环的就返回false,否则就将该结果insert进unordered_set中。
题解:
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> result;
while(1){
int sum = getSum(n);
if(sum == 1)
return true; //出现了1
}
if(result.find(sum) != result.end()){
return false; //出现了循环的结果
}else{
result.insert(sum); //将结果插入数据集中
}
n = sum; //最后记得将新计算出的sum赋值给n再计算
}
}
};