LeetCode算法题:总持续时间可被 60 整除的歌曲numPairsDivisibleBy60

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

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

 

示例 1:

输入:[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:

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

提示:

1 <= time.length <= 60000
1 <= time[i] <= 500

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pairs-of-songs-with-total-durations-divisible-by-60
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路1:
暴力破解。

思路2:

满足条件的元素对即为元素取余60后相加等于60,所以可以将所有元素取余60后存入HashMap中,利用hashmap寻找符合条件的元素对。

public int numPairsDivisibleBy60(int[] time) {

        HashMap map = new HashMap<>();
        int sum = 0;
        int a = 0,b = 0;

        for (int i = 0; i < time.length; i++) {
            int tmp = time[i] % 60;
                if(tmp == 0) {
                    a++;
                    continue;
                }
                if(tmp == 30) {
                    b++;
                    continue;
                }
            if(map.containsKey(60-tmp)){
                sum += map.get(60-tmp);
            }
            if(map.containsKey(tmp)){
                map.put(tmp,map.get(tmp)+1);
            }else{
                map.put(tmp,1);
            }
        }

        if(a != 0){
            sum += a*(a-1)/2;
        }
        if(b != 0){
            sum += b*(b-1)/2;
        }
        return sum;
    }

思路3:
对方法2的改进,将hashmap改为数组。
构造一个新的容量为60数组,将取余60的结果存入数组中,用来表示每个元素有几个,其与与hashmap思路2类似;

public int numPairsDivisibleBy60(int[] time) {
        int[] arr = new int[60];
        int sum = 0;

        for (int i = 0; i < time.length; i++) {
            arr[time[i] % 60]++;
        }

        if(arr[0] > 0){
            sum += arr[0] * (arr[0]-1) / 2;
        }

        if(arr[30] > 0){
            sum += arr[30] * (arr[30]-1) / 2;
        }

        for(int i = 1;i < 30;i++){
            sum += arr[i] * arr[60 - i];
        }
        return sum;
    }

你可能感兴趣的:(LeetCode)