题目描述
给你一个整数数组 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] < 216
二进制枚举子集
二进制表示中,某些位置才可以是1,而其它位置不能是1
如该数是 (100111)2 ,那么满足要求的二元组可以是 (100010)2或者 (000110)2,但不能是 (010001)2 。
二进制枚举子集的步骤:
- 记该数为x。我们用 sub 表示当前枚举到的子集。初始时 sub=x,因为 x 也是本身的子集;
- 我们不断地令 sub=(sub−1) & x,其中 &表示按位与运算。这样我们就可以从大到小枚举 x 的所有子集。当 sub=0时枚举结束。
方法
首先一个O(n2)的循环,列举出所有的nums[i] & nums[j] 的结果,结果在216范围内,用一个216的数组存储每个结果出现的次数,再遍历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;
}
};