一 题目:
编写一个算法来判断一个数 n
是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
- 然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
- 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n
是 快乐数 就返回 true
;不是,则返回 false
。
示例 1:
输入:n = 19 输出:true 解释: 12 + 92 = 82 82 + 22 = 68 62 + 82 = 100 12 + 02 + 02 = 1
二 思路:
根据题目分析,如果有重复的数就会出现循环就不可能最后变成1,类似于一个周期例如sinx函数不论x怎么变函数值始终在-1和1之间循环。换句话说就是出现重复的结果就不是快乐数,没有重复就一直计算直到出现1。抓住这个思路就可以用哈希算法来解决。
1.先检查sum是否为1,为1返回true
2.搜索nums_set中是否出现重复的sum,出现重复返回false
3.没有出现sum就将sum插入到nums_set中
三 代码:
#include <bits/stdc++.h>
using namespace std;
//取数各个位上的平方和
int getNum(int n){
int sum=0;
while(n){
sum+=(n%10)*(n%10);
n/=10;
}
return sum;
}
bool isHappy(int n){
unordered_set<int> nums_set;//创建一个set容器
while(true){
int sum= getNum(n);
//sum等于1就返回true
if(sum==1){
return true;
}
//先检查nums_set中有没有出现重复的sum
if(nums_set.find(sum)!=nums_set.end()){
return false;
}else{
nums_set.insert(sum);//没有重复的sum就插入到nums_set中
}
//把n变成sum
n=sum;
}
}
int main() {
int a;
cin>>a;
cout << boolalpha << isHappy(a);
return 0;
}
四 小结
如果有重复的数就会出现循环就不可能最后变成1,类似于一个周期例如sinx函数不论x怎么变函数值始终在-1和1之间循环。
当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法。
但是哈希法也是牺牲了空间换取了时间,因为我们要使用额外的数组,set或者是map来存放数据,才能实现快速的查找。