题目描述
编写一个算法来判断一个数 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
解题思路
这里考察知识点 哈希表
,
这里就要一直判断(死循环),该数的各个位置的平方和是否为 1
,如果是 1
,返回 True
如果不是 1
,可以定义一个 集合
(不能包括重复元素),判断该值是否在集合中,如果在返回 False
,如果不在,添加到集合中
代码
(C++)
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> set; // 定义集合
// 死循环
while (1) {
n = getSum(n);
if (n == 1) {
return true;
}
if (set.find(n) != set.end()) {
return false;
} else {
set.emplace(n);
}
}
}
};
(Python)
class Solution:
def getSum(self, n):
Sum = 0
while n:
Sum += (n % 10) ** 2
n = n // 10
return Sum
def isHappy(self, n: int) -> bool:
results = set() # 定义集合
while True:
n = self.getSum(n)
if n == 1:
return True
if n in results:
return False
else:
results.add(n)