C++刷leetcode（982. 按位与为零的三元组）【map】

糖逗

题目描述：

给定一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/triples-with-bitwise-and-equal-to-zero
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    int countTriplets(vector<int>& A) {
        //map
        int res = 0;
        map<int, int>store;
        int len = A.size();
        for(int i = 0; i < len; i++){
            for(int j = 0; j < len; j++){
                int temp = A[i] & A[j];
                store[temp]++;
            }
        }
        for(auto iter = store.begin(); iter != store.end(); iter++){
            if (iter -> first == 0) {
                res += len * iter -> second;
                continue;
            }
            for(int i = 0; i < len; i++){
                if((A[i] & iter -> first) == 0){
                    res += iter -> second;
                }
            }
        }
        return res;
    }
};

参考链接：https://leetcode-cn.com/problems ... yuan-zu-by-lenn123/

糖逗

昨非