# 「力扣」第 1010 题:总持续时间可被 60 整除的歌曲(中等)

# 题目描述

在歌曲列表中,第 i 首歌曲的持续时间为 time[i] 秒。

返回其总持续时间(以秒为单位)可被 60 整除的歌曲对的数量。形式上,我们希望下标数字 ij 满足 i < j 且有 (time[i] + time[j]) % 60 == 0

示例 1:

输入:time = [30,20,150,100,40]
输出:3
解释:这三对的总持续时间可被 60 整除:
(time[0] = 30, time[2] = 150): 总持续时间 180
(time[1] = 20, time[3] = 100): 总持续时间 120
(time[1] = 20, time[4] = 40): 总持续时间 60

示例 2:

输入:time = [60,60,60]
输出:3
解释:所有三对的总持续时间都是 120,可以被 60 整除。

提示:

  • 1 <= time.length <= 6 * 10^4
  • 1 <= time[i] <= 500

# 思路分析

思路有点像 「力扣」第 1 题: 两个数之和 (opens new window)「力扣」第 454 题:四数相加 II (opens new window)

1、题目要求找整数对,并且是“总持续时间”,即两首歌曲之和,“可被 60 整除”,因此我们做预处理,把数组里所有的数处理成模 60 的余数;

说明:这个思路来自「力扣」第 1 题: 两数之和。

2、做计数统计是必要的,因此需要使用哈希表,键是当前遍历的秒数,值是这个秒数出现的次数,在遍历的过程中,统计 “60 - 当前遍历的秒数”,这个秒数已经出现了几次,那么当前遍历的数就能够和之前的若干次组成“歌曲对”;

说明:这个思路来自「力扣」第 18 题: 四数之和。

3、针对整一分钟,例如 [60, 60, 60] 这样的测试用例,预处理完成以后就是 [0, 0, 0],所以 “60 - 当前遍历的秒数” 也要模 60。

说明:这是在提交测试用例的时候发现的。

参考代码

Java 代码:

import java.util.HashMap;
import java.util.Map;

public class Solution {

    public int numPairsDivisibleBy60(int[] time) {
        // 预处理:把数组中的元素全都模 60
        int len = time.length;
        for (int i = 0; i < len; i++) {
            time[i] = time[i] % 60;
        }
        // 注意:[60, 60, 60] 会被处理成 [0, 0, 0]
        Map<Integer, Integer> map = new HashMap<>();
        int res = 0;
        for (int i = 0; i < len; i++) {
            // 注意:要记得模 60
            int residue = (60 - time[i]) % 60;

            Integer preCount = map.get(residue);
            if (preCount != null) {
                res += preCount;
            }

            // 计数,应该放在统计之后做,因为后面的参考前面的
            Integer curCount = map.get(time[i]);
            if (curCount == null) {
                map.put(time[i], 1);
            } else {
                map.put(time[i], curCount + 1);
            }
        }
        return res;
    }

}

Python 代码:

from typing import List


class Solution:

    def numPairsDivisibleBy60(self, time: List[int]) -> int:
        # 预处理:把数组中的元素全都模 60
        time = [t % 60 for t in time]

        from collections import defaultdict
        d = defaultdict(int)

        res = 0
        for t in time:
            # 1、先计数
            # 针对 [0, 0, 0] 这一类特殊用例,要模 60
            residue = (60 - t) % 60
            if residue in d:
                res += d[residue]

            # 2、再记录频数
            d[t] += 1

        return res

复杂度分析:

  • 时间复杂度:,这里 是数组的长度,算法把数组看了两次;
  • 空间复杂度:,使用了长度为 的哈希表。

作者:liweiwei1419 链接:https://suanfa8.com/hash-table/solutions/1010-pairs-of-songs-with-total-durations-divisible-by-60 来源:算法吧 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

Last Updated: 11/19/2024, 7:59:29 AM