力扣刷题记录#数组#简单#1010总持续时间可被 60 整除的歌曲

题目描述

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

示例

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

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

解答

class Solution(object):
    def numPairsDivisibleBy60(self, time):
        """
        :type time: List[int]
        :rtype: int
        """
        
        # remainder存储余数的计数
        # 即remainder[0]表示余数为0的时间的数量,remainder[20]表示余数为20的时间的数量
        remainder = [0]*60
        for num in time:
            remainder[num%60] = remainder[num%60] + 1
            
        ans = 0
        for i in range(1,30):
            # 余数为i和余数为60-i的组合,才能被60整除
            ans += remainder[i] * remainder[60-i]
        
        # 余数为0和余数为30是特殊情况
        ans += (remainder[0]-1) * remainder[0] / 2
        ans += (remainder[30]-1) * remainder[30] / 2
        
        return ans

你可能感兴趣的:(力扣)