在 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