编写一个算法来判断一个数 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会重复出现,这对解题很重要!
正如:关于哈希表,你该了解这些! (opens new window)中所说,当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。
所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。
判断sum是否重复出现就可以使用unordered_set。
还有一个难点就是求和的过程,如果对取数值各个位上的单数操作不熟悉的话,做这道题也会比较艰难。
public:
int getSum(int n) {
int sum = 0;
while (n) {
//在循环体中,(n % 10) 可以提取 n 的最后一位数字
sum += (n % 10) * (n % 10);
//通过 n /= 10 操作,将处理过的数字从 n 中去除,相当于把 n 的最后一位删除。
n /= 10;
}
return sum;
}
bool isHappy(int n) {
unordered_set<int> set;
while (1) {
int sum = getSum(n);
//如果sum等于1,是快乐数,直接返回true;
if (sum == 1) {
return true;
}
//如果sum在set中可以找到,说明已经陷入循环,函数返回 false,因为不可能得到1了。
if (set.find(sum) != set.end()) {
return false;
}
//如果set中没有包含当前的sum,就将sum插入set中;
else {
set.insert(sum);
}
//将 n 更新为 sum,继续下一轮循环,重新计算平方和。
n = sum;
}
}
};