982. 按位与为零的三元组

题目描述

给你一个整数数组 nums ，返回其中 按位与三元组 的数目。
按位与三元组 是由下标 (i, j, k) 组成的三元组，并满足下述全部条件：
0 <= i < nums.length
0 <= j < nums.length
0 <= k < nums.length
nums[i] & nums[j] & nums[k] == 0 ，其中 & 表示按位与运算符。

示例 1：
输入：nums = [2,1,3]
输出：12
解释：可以选出如下 i, j, k 三元组：
(i=0, j=0, k=1) : 2 & 2 & 1
(i=0, j=1, k=0) : 2 & 1 & 2
(i=0, j=1, k=1) : 2 & 1 & 1
(i=0, j=1, k=2) : 2 & 1 & 3
(i=0, j=2, k=1) : 2 & 3 & 1
(i=1, j=0, k=0) : 1 & 2 & 2
(i=1, j=0, k=1) : 1 & 2 & 1
(i=1, j=0, k=2) : 1 & 2 & 3
(i=1, j=1, k=0) : 1 & 1 & 2
(i=1, j=2, k=0) : 1 & 3 & 2
(i=2, j=0, k=1) : 3 & 2 & 1
(i=2, j=1, k=0) : 3 & 1 & 2

示例 2：
输入：nums = [0,0,0]
输出：27

提示：
1 <= nums.length <= 1000
0 <= nums[i] < 2¹⁶

二进制枚举子集

二进制表示中，某些位置才可以是1，而其它位置不能是1
如该数是 (100111)₂ ，那么满足要求的二元组可以是 (100010)₂或者 (000110)₂，但不能是 (010001)₂ 。
二进制枚举子集的步骤：

• 记该数为x。我们用 sub 表示当前枚举到的子集。初始时 sub=x，因为 x 也是本身的子集；
• 我们不断地令 sub=(sub−1) & x，其中 &表示按位与运算。这样我们就可以从大到小枚举 x 的所有子集。当 sub=0时枚举结束。

方法

首先一个O(n²)的循环，列举出所有的nums[i] & nums[j] 的结果，结果在2¹⁶范围内，用一个2¹⁶的数组存储每个结果出现的次数，再遍历nums[k]，由于前面的结果跟nums[k]相与为0，则nums[k]为1的地方必须是0，nums[k]为0的地方可以为1，令x为跟nums[k]按位异或的结果，求出x的二进制枚举子集即可。

代码

class Solution {
public:
    int countTriplets(vector<int>& nums) {
        vector<int> cnt(1 << 16);
        for (int x: nums) {
            for (int y: nums) {
                ++cnt[x & y];
            }
        }
        int ans = 0;
        for (int x: nums) {
            x = x ^ 0xffff;
            for (int sub = x; sub; sub = (sub - 1) & x) {
                ans += cnt[sub];
            }
            ans += cnt[0];
        }
        return ans;
    }
};