leetcode 982

Posted 2021-11-30 bella2017

题意：寻找一个整数数组A中的三个数，使得它们与为0

思路：使用 unordered_map , key键存储两层for循环后得到的与值，再将unordered_map的所有key值与A里的所有值相与，若为0则将 A.second 加到cnt中。

class Solution {
public:
    int countTriplets(vector<int>& A) {
        int cnt = 0;
        unordered_map<int , int> tri;
        for(auto i:A)
            for(auto j : A)
                ++tri[i&j];
        for(auto k:A)
            for(auto t:tri)
                if((k & t.first) == 0)
                    cnt += t.second;
        return cnt;
    }
};

时间复杂度分析：因为A[i]的最大值为2^16，所以map的最多存储2^16个数，时间复杂度为  O(n*n)