在 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 许愿星姐妹祝福语简短