1010. 总持续时间可被 60 整除的歌曲

2019-05-12  本文已影响0人  好吃红薯

在歌曲列表中,第 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

思路:拿道题一看,O(n²)的版本,果然超时

class Solution:
    def numPairsDivisibleBy60(self, time: List[int]) -> int:
        res = 0
        for i in range(len(time)-1):
            for j in range(i+1,len(time)):
                if (time[i]+time[j])%60==0:
                    res+=1
        return res

思路:余数相加为60的两个数组合,可以被60整除

class Solution:
    def numPairsDivisibleBy60(self, time: List[int]) -> int:
        res = 0
        map ={}
        for i in range(60):
            map[i] = 0
        
        for i in time:
            map[i%60] += 1
        
        #余数为0
        res += map[0]*(map[0]-1)//2

        #余数1-29 对应59-31的组合
        for i in range(1,30):
            res += map[i]*map[60-i]

        #余数30
        res += map[30]*(map[30]-1)//2

        return res
上一篇下一篇

猜你喜欢

热点阅读