「快乐数」 定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
输入:n = 19
输出:true
解释:
1² + 9² = 82
8² + 2² = 68
6² + 8² = 100
1² + 0² + 0² = 1
输入:n = 2
输出:false
Python3 divmod() 函数 divmod(a, b)
如果参数 a 与 参数 b 都是整数,函数返回的结果相当于 (a // b, a % b)。
整除 余数
>>> divmod(7, 2)
(3, 1)
>>> divmod(8, 2)
(4, 0)
>>> divmod(8, -2)
(-4, 0)
>>> divmod(3, 1.3)
(2.0, 0.3999999999999999)
# 新建一个函数,用来计算每一位的平方数,然后相加
# digit 每次得到n的个位数,然后计算平方,while循环得到所有和
# 如果seen中有重复的书,说明进入循环
class Solution:
def isHappy(self, n: int) -> bool:
def get_next(n):
total_sum = 0
while n>0:
n,digit = divmod(n,10)
total_sum += digit ** 2
return total_sum
seen = set()
while n!= 1 and n not in seen:
seen.add(n)
n= get_next(n)
return n==1
算法分为两部分,我们需要设计和编写代码。
1. 给一个数字 n,它的下一个数字是什么?
2 . 按照一系列的数字来判断我们是否进入了一个循环。
第 1 部分我们按照题目的要求做数位分离,求平方和。
第 2 部分可以使用哈希集合完成。每次生成链中的下一个数字时,我们都会检查它是否已经在哈希集合中。
如果它不在哈希集合中,我们应该添加它。
如果它在哈希集合中,这意味着我们处于一个循环中,因此应该返回 false。