给定一个整数数组 A,找出索引为 (i, j, k)
的三元组,使得:
0 <= i < A.length
0 <= j < A.length
0 <= k < A.length
A[i] & A[j] & A[k] == 0,其中 & 表示按位与(AND)操作符。
示例:
输入:[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
提示:
1 <= A.length <= 1000
0 <= A[i] < 2^16
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/triples-with-bitwise-and-equal-to-zero 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
int countTriplets(vector<int>& A) {
vector<int> count(1<<16, 0);//两个数 & 操作后,不会变大
for(int i = 0; i < A.size(); ++i)
{
for(int j = 0; j < A.size(); ++j)
{
count[A[i] & A[j]]++;
}
}
int ans = 0;
for(int i = 0; i < A.size(); ++i)
{
for(int j = 0; j < (1<<16); ++j)
{
if((A[i] & j) == 0)
ans += count[j];
}
}
return ans;
}
};
时间复杂度 O ( 2 16 n ) O(2^{16}n) O(216n)
648 ms 14.7 MB C++