cf769D(枚举&位或运算)

题目链接：http://codeforces.com/problemset/problem/769/D

 

题意：求给出的 n 个数中有多少对数字的二进制形式恰好有 k 位不同

 

思路：两个数a, b的二进制形式恰好 k 位不同即 a ^ b 中 1 的个数，那么可以枚举．注意 n 为 1e5 枚举 ai 会tle，不过 ai 的范围是 1e4，显然有大量重复的元素；

可以先去重并记录每个元素出现的次数，再枚举每个元素即可；

 

代码：

#include <iostream>
#include <stdio.h>
#define ll long long
using namespace std;

const int MAXN = 1e5 + 10;
ll a[MAXN], vis[MAXN];

int BitCount(unsigned int n){//求 n 二进制中 1 的数目
    unsigned int tmp = n - ((n >> 1) & 033333333333) - ((n >> 2) & 011111111111);
    return ((tmp + (tmp >> 3)) & 030707070707) % 63;
}

int main(void){
    int n, k, x, indx = 0;
    ll ans = 0;
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i++){
        scanf("%d", &x);
        if(!vis[x]) a[indx++] = x;
        vis[x]++;
    }
    for(int i = 0; i < indx; i++){
        if(k == 0){
            ans += (vis[a[i]] - 1) * vis[a[i]] / 2;
            continue;
        }
        for(int j = i + 1; j < indx; j++){
            int cnt = a[i] ^ a[j];
            if(k == BitCount(cnt)) ans += vis[a[i]] * vis[a[j]];
        }
    }
    printf("%lld\n", ans);
    return 0;
}