Codeforces 769D

太久没写搜索因为递归边界问题卡了很久。。

题意：定义k-interesting：如果两个数的二进制形式有k位不相同，则称之为k-interesting。给出n和k，输入n个大小在[0,10000]之间的数，问有多少对ai,aj（i<j）是k-interesting数。

解题思路：注意到n个数的大小特地限制在10000以下，考虑暴力搜索，复杂度为10000*(2^14)，复杂度太高，需要剪枝：当k小于7，计算k位不相同的数有哪些；k大于等于7，计算14-k位相同的数有哪些。使用vector存储搜索结果，遍历一遍即可（结果需用long long存储，极端情况下答案为n^2级别）

代码如下：（表现十分不佳——1699ms，大概还是姿势太挫了。。？？）

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
typedef long long ll;
#define sqr(x) ((x)*(x))
const int N=1e4+;
int n,k,a[N*],tot[N*];
vector<int>all[N];
void dfs(int x,int num,int pos,int cnt,int rev){
    if(cnt<) return;
    if(pos==-){
        if(!cnt){
            all[x].push_back(num);
            return;
        }
        return;
    }
    int tmp=<<pos;
    if(num&tmp) tmp*=-;
    num+=tmp;
    dfs(x,num,pos-,rev?cnt:cnt-,rev);
    num-=tmp;
    dfs(x,num,pos-,rev?cnt-:cnt,rev);
}
void init(){
    int rev=k>=?:;
    if(rev) k=-k;
    for(int i=;i<=;i++){
        all[i].clear();
        dfs(i,i,,k,rev);
    }
    memset(tot,,sizeof(tot));
}
int main(){
    //freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&k)){
        init();
        for(int i=;i<n;i++)
            scanf("%d",a+i);
        ll res=;
        for(int i=n-;i>=;i--){
            for(int j=;j<all[a[i]].size();j++) if(tot[all[a[i]][j]]){
                res+=tot[all[a[i]][j]];
            }
            tot[a[i]]++;
        }
        printf("%I64d\n",res);
    }
    return ;
}