本题使用了哈希表中的 unordered_set 数据结构,关键点在于判断出平方和的重复出现导致无限循环始终变不到 1 这个条件。
时间复杂度:O(log n),空间复杂度:O(log n)
思路:
- 使用 while 循环来进行迭代计算,每次迭代会将数字 n 的每一位平方相加得到新的数字 sum。如果 sum 等于 1,则说明输入的数字是快乐数,函数返回 true。
-
如果 sum 不等于 1,代码使用一个 unordered_set 来记录已经出现过的数字,如果出现了重复的数字,则说明数字不是快乐数,函数返回 false。
-
如果未出现重复数字,则将 sum 赋值给 n,并将 sum 重置为 0,继续迭代计算。
代码:
class Solution {
public:
bool isHappy(int n) {
unordered_set<int> set;
int sum = 0;
while (1) {
while (n != 0) {
sum += pow(n % 10, 2);
n /= 10;
}
if (sum == 1) {
return true;
} else {
if (set.find(sum) != set.end()) {
return false;
} else {
n = sum;
set.insert(sum);
sum = 0;
}
}
}
}
};