在 Python 中计算元素总和为 2 的幂的索引对的程序
假设我们有一个名为nums的数字列表。我们必须找到索引对i,j的数量,其中i<j使得nums[i]+nums[j]对于某些0>=k等于2^k。
因此,如果输入类似于nums=[1,2,6,3,5],那么输出将是3,因为有三对sum,如(6,2):sumis8,(5,3):sum是8并且(1,3)sum是4
示例
让我们看下面的实现来更好地理解
from collections import Counter
def solve(nums):
res, c = 0, Counter()
for x in nums:
for j in range(32):
res += c[(1 << j) - x]
c[x] += 1
return res
nums = [1, 2, 6, 3, 5]
print(solve(nums))输入
[1, 2, 6, 3, 5]输出结果
3
热门推荐
10 小红书平安祝福语简短
11 生日祝福语大全女孩简短
12 收生日红包祝福语 简短
13 领证幽默祝福语简短
14 法考面试祝福语简短
15 老哥出门祝福语简短语
16 送灯祝福语简短独特
17 幼儿狗年祝福语大全简短
18 好听的元旦简短祝福语