Codeforces 769D k-Интересные пары чисел

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

---

搜索题

考虑这些数的值域较小，直接${O(2^{k})}$次方枚举每个数字二进制位上是否改变,剪枝一下最不利复杂度为${O(C_{14}^{7})}$，有${10^{4}}$中取值。

最终最不利复杂度为${O(C_{14}^{7}*(10^{4}))}$

---

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
#define maxn 100010
#define llg long long 
#define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
llg n,m,str[maxn],c[maxn],k,v[maxn];
vector<llg>a[maxn];
llg d[maxn];
void ss1(llg wz,llg res,llg x)
{
    if (wz==14)
    {
        if (res==0)
        {
            llg val=0;
            for (llg i=0;i<14;i++) val+=(1<<i)*((str[i]+c[i])%2);
            a[x].push_back(val);
        }
        return ;
    }
    if (res)
    {
    c[wz]=1;
    ss1(wz+1,res-1,x);
    }
    c[wz]=0;
    ss1(wz+1,res,x);
}

void ss2(llg wz,llg res,llg x)
{
    if (wz==14)
    {
        if (res==0)
        {
            llg val=0;
            for (llg i=0;i<14;i++) val+=(1<<i)*((str[i]+c[i])%2);
            a[x].push_back(val);
        }
        return ;
    }
    if (res)
    {
        c[wz]=0;
        ss2(wz+1,res-1,x);
    }
    c[wz]=1;
    ss2(wz+1,res,x);
}

void change_(llg x)
{
    llg cs=0;
    while (cs<14)
    {
        str[cs]=x%2;
        x/=2;
        cs++;
    }
}

int main()
{
    yyj("D");
    cin>>n>>k;
    for (llg i=0;i<=10000;i++)
    {
        change_(i);
        if (k<=7) ss1(0,k,i);else ss2(0,14-k,i);
    }
    for (llg i=1;i<=n;i++) scanf("%lld",&v[i]),d[v[i]]++;
    llg ans=0;
    for (llg i=1;i<=n;i++)
    {
        llg w=a[v[i]].size(),x=v[i];
        for (llg j=0;j<w;j++) 
            if (k==0) ans+=d[a[x][j]]-1; else ans+=d[a[x][j]];
    }
    cout<<ans/2;
    return 0;
}