Triples with Bitwise AND Equal To Zero

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. 1 <= A.length <= 1000
2. 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];
    }
}