【LeetCode】力扣刷题之路
题目(20240527~20240531)
202.快乐数
编写一个算法来判断一个数 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 <= n <= 231 - 1
方法1:哈希集合检测循环
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
if __name__ == "__main__":
s = Solution()
# 202.快乐数 - 示例1
n = 19
res = s.isHappy(n)
print(res) # 输出:true
# 示例2
n = 2
res = s.isHappy(n)
print(res) # 输出:false
方法2:快慢指针
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
slow = n
fast = get_next(n)
while fast != 1 and slow != fast:
slow = get_next(slow)
fast = get_next(get_next(fast))
return fast == 1
if __name__ == "__main__":
s = Solution()
# 202.快乐数 - 示例1
n = 19
res = s.isHappy(n)
print(res) # 输出:true
# 示例2
n = 2
res = s.isHappy(n)
print(res) # 输出:false
888.公平的糖果交换
爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ,aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量,bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。
两人想要互相交换一盒糖果,这样在交换之后,他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。
返回一个整数数组 answer,其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目,answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案,你可以返回其中任何一个。题目测试用例保证存在与输入对应的答案。
示例 1:
输入:aliceSizes = [1,1], bobSizes = [2,2]
输出:[1,2]
示例 2:
输入:aliceSizes = [1,2], bobSizes = [2,3]
输出:[1,2]
示例 3:
输入:aliceSizes = [2], bobSizes = [1,3]
输出:[2,3]
示例 4:
输入:aliceSizes = [1,2,5], bobSizes = [2,4]
输出:[5,4]
提示:
- 1 <= aliceSizes.length, bobSizes.length <= 104
- 1 <= aliceSizes[i], bobSizes[j] <= 105
- 爱丽丝和鲍勃的糖果总数量不同。
- 题目数据保证对于给定的输入至少存在一个有效答案。
代码
from typing import List
class Solution:
def fairCandySwap(self, aliceSizes: List[int], bobSizes: List[int]) -> List[int]:
suma = sum(aliceSizes)
sumb = sum(bobSizes)
delta = (suma - sumb) // 2
alice = set(aliceSizes)
res = None
for y in bobSizes:
x = y + delta
if x in alice:
res = [x,y]
break
return res
if __name__ == '__main__':
s = Solution()
# 888.公平的糖果交换 - 示例1
aliceSizes = [1, 1]
bobSizes = [2, 2]
res = s.fairCandySwap(aliceSizes, bobSizes)
print(res) # 输出:[1,2]
# 示例2
aliceSizes = [1, 1]
bobSizes = [2, 3]
res = s.fairCandySwap(aliceSizes, bobSizes)
print(res) # 输出:[1,2]
# 示例3
aliceSizes = [2]
bobSizes = [1, 3]
res = s.fairCandySwap(aliceSizes, bobSizes)
print(res) # 输出:[2,3]
# 示例4
aliceSizes = [1, 2, 5]
bobSizes = [2, 4]
res = s.fairCandySwap(aliceSizes, bobSizes)
print(res) # 输出:[5,4]