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