Given an array of integers A
, find the number of triples of indices (i, j, k) such that:
0 <= i < A.length
0 <= j < A.length
0 <= k < A.length
A[i] & A[j] & A[k] == 0
, where&
represents the bitwise-AND operator.
Example 1:
Input: [2,1,3]
Output: 12
Explanation: We could choose the following i, j, k triples:
(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
Note:
1 <= A.length <= 1000
0 <= A[i] < 2^16
题目理解:
定义三元组:在数组中任意取出三个数字,可以重复,这三个数字相与的结果为0。给定一个数组,求其中三元组的数量。
解题思路:
从约束中可以看出,数字的大小是2^16,也就是说,最大的数字不会超过65535,因此我们可以申请一个65536大小的数组nums,nums[i]表示i出现的次数。将A中的数字全部加入到nums中。这一过程执行三次,每一次加入一个数字的时候,都和已经存在与nums中的数字进行与操作,这样,在执行三次之后,nums中的每一个下标i都是A中选出三个数字进行与操作的结果。
class Solution {
public int countTriplets(int[] A) {
int[] cur = new int[65536];
for(int num : A)
cur[num]++;
for(int count = 1; count < 3; count++){
int[] next = new int[65536];
for(int num : A){
for(int i = 0; i < 65536; i++){
if(cur[i] == 0)
continue;
next[num & i] += cur[i];
}
}
cur = next;
}
return cur[0];
}
}