【LeetCode】力扣刷题之路 (20240527~20240531)

题目(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 <= 2³¹ - 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 <= 10⁴
• 1 <= aliceSizes[i], bobSizes[j] <= 10⁵
• 爱丽丝和鲍勃的糖果总数量不同。
• 题目数据保证对于给定的输入至少存在一个有效答案。

代码

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]