【Leetcode】2354. Number of Excellent Pairs

题目地址：

https://leetcode.com/problems/number-of-excellent-pairs/description/

给定一个长$n$数组$A$，其中两个数$x,y$如果满足$x\&y$和$x\wedge y$的二进制位的$1$的总个数大于等于某个给定的正整数$k$，则说这两个数形成了一个好数对。问好数对的总个数。一个数对可以看做是在$A$中可重复地挑出两个数所组成的，同一个数可以被挑多次。两个数对不同当且仅当其某一维不同。

首先将$A$去个重。$x\&y$和$x\wedge y$的二进制位的$1$的总个数其实就等于两个数的二进制位$1$的总个数。可以统计一下二进制位$1$的总个数为$b$的数字有多少个，得数组$c$（即$c[b]$为二进制位$1$的个数是$b$的数字个数），那么答案就是$$\sum _{i+j\ge k}c[i]c[j]$$代码如下：

class Solution {
 public:
  using ll = long long;
  ll countExcellentPairs(vector<int>& a, int k) {
    sort(a.begin(), a.end());
    a.erase(unique(a.begin(), a.end()), a.end());

    auto f = [](int x) {
#define lowbit(x) (x) & (-x)
      int cnt = 0;
      while (x) x -= lowbit(x), cnt++;
      return cnt;
    };

    vector<int> cnt(31, 0);
    for (int x : a) cnt[f(x)]++;

    ll res = 0;
    for (int i = 0; i < 31; i++)
      for (int j = max(0, k - i); j < 31; j++) res += (ll)cnt[i] * cnt[j];
    return res;
  }
};

时间复杂度$O(n\log n)$，空间$O(1)$。