[leetcode] 982. 按位与为零的三元组

题目

给定一个整数数组 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. 1 <= A.length <= 1000
2. 0 <= A[i] < 2^16

解题思路：

暴力法… 所以我也不知道为什么是hard其实

代码

class Solution {
public:
    int countTriplets(vector<int>& A) {
        map<int, int> freq;
        for(int i = 0; i < A.size(); i++){
            for(int j = 0; j < A.size(); j++){
                int a = A[i] & A[j];
                if(freq.find(a) == freq.end()){
                    freq[a] = 1;
                }
                else{
                    freq[a] += 1;
                }
            }
        }

        int res = 0;
        for(auto iter = freq.begin(); iter != freq.end(); iter++){
            for(int i = 0; i < A.size(); i++){
                if(((iter->first) & A[i]) == 0){
                    res += (iter->second);
                }
            }
        }
        return res;
    }
};