给定一个整数数组 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
思路:我们将任意两个元素与的结果存入map中,并将结果与结果出现的次数进行映射,之后我们遍历这个map表,通过查看原数组中有多少个元素与map相应的值与的结果为0,从而更新答案。
class Solution {
public int countTriplets(int[] A) {
int ans=0;
int n=A.length;
int[] map=new int[1<<16];
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
map[A[i]&A[j]]++;
for(int i=0;i<map.length;i++) {
if(map[i]==0) continue;
for(int j=0;j<n;j++)
if((A[j]&i)==0)
ans+=map[i];
}
return ans;
}
}