在歌曲列表中,第 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
代码实现
C++:
这个问题很简单,我们只需要知道一个概念:
对于整数 i, j,如果i+j可被60整除,则存在如下的等价关系:
(i+j)/60 等价于 (i%60 + j%60) == 60
这样我们就可以将任意值的整数a和b限制在0~59之间,因为它们的余数不会超过59。
因此,我们可以建立一个长度是60的数组rem,下标059对应059的余数,每个元素的值记录出现该余数的歌曲的数量。
我们对数组time进行遍历,对于每一个歌曲时长i:
-
计算出与60余运算后的余数index = i%60;
-
这样如果rem[60-index]不为空,那么里面记录的每一个歌曲总时长都可以和i相加后得到的总持续时间都能被60整除;
注意我们要考虑一种特殊情况:
当i可以被60整除时,index是0,那么这时候我们期望找到其他的可以被60整除的歌曲时长,就依然是rem[0],所以我们将60-index要写成(60-index)%60;
- 最后我们还要将当前歌曲记录到rem中的对应位置,以使得后续的歌曲可以与之配对。
class Solution {
public:int numPairsDivisibleBy60(vector<int>& time) {int count=0, rem[60] = {0};for(int i=0; i<time.size(); i++){int index = time[i]%60;//(i+j)/60 等价于 (i%60 + j%60) == 60count += rem[(60-index)%60];rem[index]++;}return count;}
};
Python:
class Solution(object):def numPairsDivisibleBy60(self, time):""":type time: List[int]:rtype: int"""rem = [0]*60count = 0for i in time:index = i%60count += rem[(60-index)%60]rem[index] += 1return count