是否有一种有效的方法来计算总计整分钟的对数

Question

[我正在尝试查找持续时间加起来长达一整分钟的成对歌曲。给定示例歌曲长度[10、50、90、30]。计算不同对的总数。我期望自第一对和第二对达到60秒，第三和第四对达到120秒以来返回2。

def pair_with_target_sum(songs, k):
    n = len(songs)
    count = 0
    for i in range(0, n):
        for j in range(i + 1, n):
            if songs[i] + songs[j] == k:
                count += 1
    return count

def main():
    print(pair_with_target_sum([10, 50, 90, 30], 60))
    print(pair_with_target_sum([30, 20, 150, 100, 40], 60))

main()

[我正在尝试查找持续时间加起来长达一整分钟的成对歌曲。给定示例歌曲长度[10、50、90、30]。计算不同对的总数。我期望返回2 ...

Answer 1

有不同且更简单的算法：

Answer 2

我将itertools.combinations与itertools.combinations结合使用：