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 + ;
 ll a[MAXN], vis[MAXN];

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

 int main(void){
     int n, k, x, indx = ;
     ll ans = ;
     scanf("%d%d", &n, &k);
     for(int i = ; i < n; i++){
         scanf("%d", &x);
         if(!vis[x]) a[indx++] = x;
         vis[x]++;
     }
     for(int i = ; i < indx; i++){
         if(k == ){
             ans += (vis[a[i]] - ) * vis[a[i]] / ;
             continue;
         }
         for(int j = i + ; 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 ;
 }