用Python解决LeetCode 1010：划分总数能被60整除的歌曲对

2023-12-13 23:12:01

前言

LeetCode 1010题要求我们计算，给定一组歌曲时长，能组成多少对歌曲的总时长可以被60整除。为了解决这个问题，我们将介绍两种方法：暴力枚举法和使用哈希表进行优化的方法。

2. 暴力枚举法

暴力枚举法是最简单直接的解决方案。我们可以枚举每一对歌曲，并计算它们的总时长是否能被60整除。如果能被整除，我们就将这组歌曲对计入结果。

def numPairsDivisibleBy60(time):
    """
    :type time: List[int]
    :rtype: int
    """
    # 初始化结果计数器
    count = 0

    # 暴力枚举每一对歌曲
    for i in range(len(time)):
        for j in range(i+1, len(time)):
            # 计算歌曲对的总时长
            total_time = time[i] + time[j]

            # 检查总时长是否能被60整除
            if total_time % 60 == 0:
                # 如果能被整除，则增加计数器
                count += 1

    # 返回结果
    return count

3. 哈希表优化法

第二种方法使用哈希表来优化计算。哈希表是一种数据结构，它允许我们快速查询和修改数据。在我们的问题中，我们可以使用哈希表来存储每个歌曲时长及其余数。当我们枚举每一对歌曲时，我们可以通过哈希表快速查找出另一首歌曲的余数，从而判断它们是否能组成总时长能被60整除的歌曲对。

def numPairsDivisibleBy60(time):
    """
    :type time: List[int]
    :rtype: int
    """
    # 初始化哈希表
    hash_table = {}

    # 初始化结果计数器
    count = 0

    # 遍历歌曲时长数组
    for song_time in time:
        # 计算歌曲时长的余数
        remainder = song_time % 60

        # 如果余数是0，则可以与任何歌曲组成有效对
        if remainder == 0:
            count += hash_table.get(0, 0)

        # 如果余数不是0，则检查哈希表中是否存在与之互补的余数
        else:
            complementary_remainder = 60 - remainder
            count += hash_table.get(complementary_remainder, 0)

        # 将当前歌曲时长的余数添加到哈希表中
        hash_table[remainder] = hash_table.get(remainder, 0) + 1

    # 返回结果
    return count