【Codeforces241B】Friends

题意：
给你n个数，问从其中选择互不相同的m对数的异或和之和最大为多少。

在Trie上dp应该就好了。

#include <bits/stdc++.h>#define gc getchar()#define ll long long#define N 50009#define T 31#define mod 1000000007#define inv2 ((mod+1)/2)using namespace std;int n,m,sum[N][T];unsigned limit,a[N],number;struct Trie{    int size[N*T],root,ch[N*T][2],cnt,pos[N];    unsigned S;    void init()    {        root=cnt=1;    }    void ins(int x)    {        int now=root;        for (int i=T-1;i>=0;i--)        {            size[now]++;            int c=x>>i&1;            if (!ch[now][c]) ch[now][c]=++cnt;            now=ch[now][c];        }        size[now]++;    }    void work()    {        for (int i=1;i<=n;i++)            pos[i]=root;        number=limit=0;        for (int i=T-1;i>=0;i--)        {            S=0;            for (int j=1;j<=n;j++)                S+=size[ch[pos[j]][!((limit^a[j])>>i&1)]];            if (S+number<=m*2) number+=S;            else limit^=1<<i;            for (int j=1;j<=n;j++)                pos[j]=ch[pos[j]][(limit^a[j])>>i&1];        }        number/=2;    }    int calc(int ret=0)    {        for (int i=1;i<=n;i++)            for (int j=0;j<T;j++)                if (!(limit>>j&1))                {                    unsigned Min=a[i]^(limit|(1<<j));                    Min&=~0<<j;                    int l=lower_bound(a+1,a+n+1,Min)-a;                    int r=lower_bound(a+1,a+n+1,Min+(1<<j))-a;                    ret=(ret+(ll)((limit|(1<<j))&(~0<<j))%mod*(r-l)%mod)%mod;                    for (int k=0;k<j;k++)                    {                        int num=sum[r-1][k]-sum[l-1][k];                        if (a[i]>>k&1) num=r-l-num;                        ret=(ret+(ll)num*((1<<k)%mod)%mod)%mod;                    }                }        return (ll)ret*inv2%mod;    }}trie;int read(){    int x=1;    char ch;    while (ch=gc,ch<'0'||ch>'9') if (ch=='-') x=-1;    int s=ch-'0';    while (ch=gc,ch>='0'&&ch<='9') s=s*10ll+ch-'0';    return s*x;}int main(){    trie.init();    n=read(),m=read();    for (int i=1;i<=n;i++)        trie.ins(a[i]=read());    if (m==0)    {        puts("0");        return 0;    }    sort(a+1,a+n+1);    for (int i=0;i<T;i++)        sum[0][i]=0;    for (int i=1;i<=n;i++)        for (int j=0;j<T;j++)            sum[i][j]=sum[i-1][j]+(a[i]>>j&1);    trie.work();    printf("%d\n",(trie.calc()+(ll)(m-number)*(limit%mod)%mod)%mod);    return 0;}