从给定数组中按位或等于K的对对进行计数

简介

给定一个整数数组和一个目标整数K，任务是计算数组中位或等于K的不同对的数量。

算法思路

1. 创建一个空的集合 pairs，用于存储找到的符合条件的对。
2. 遍历给定数组中的每个元素 num：
   • 对于每个元素 num，遍历集合 pairs 中的每个对 (x, y)：
     • 如果 (x | num) == K 或者 (y | num) == K，则将该对 (x, y) 加入到集合 pairs 中。
   • 将 { num, K - num } 这对作为一个新的对 (x, y)，加入到集合 pairs 中。
3. 返回集合 pairs 中的不同对的数量。

代码示例

def count_pairs(arr, K):
    pairs = set()

    for num in arr:
        for x, y in pairs.copy():
            if (x | num) == K or (y | num) == K:
                pairs.add((x, y))

        pairs.add((num, K - num))

    return len(pairs)

# 示例用法
arr = [1, 2, 3, 4, 5]
K = 7
result = count_pairs(arr, K)
print(result)  # 输出: 3

复杂度分析

• 时间复杂度：O(N^2)，其中 N 是数组的长度。
• 空间复杂度：O(N)，需要额外的空间来存储符合条件的对。

注意：上述代码中的解决方法并非最优解，而是一种简单的暴力解法，实际应用中可能存在更优的解决方案。