【清华集训2014】【BZOJ3811】玛里苟斯

Description

魔法之龙玛里苟斯最近在为加基森拍卖师的削弱而感到伤心，于是他想了一道数学题。
S 是一个可重集合，S={a1,a2,…,an}。
等概率随机取 S 的一个子集 A={ai1,…,aim}。
计算出 A 中所有元素异或 x, 求 xk 的期望。
Input

第一行两个正整数 n, k。
以下 n 行每行一个整数，表示 ai。
Output

如果结果是整数，直接输出。如果结果是小数（显然这个小数是有限的），输出精确值（末尾不加多余的 0）。
Sample Input

4 2

0

1

2

3
Sample Output

3.5
HINT

限制与约定

1≤n≤100000，1≤k≤5，ai≥0。最终答案小于 2^63 。k=1,2,3,4,5 各自占用 20% 的数据

Source

2015年国家集训队测试

感觉这个题是清华集训2014Day1三个题里最难的..
看上去一脸不可做,实际上也确实是一脸不可做..
自己想只会做k=1的..按位统计每位的异或期望值..
安利两个很好的题解.
题解1
题解2
结合起来看,看了一上午,最后看懂了…

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 100010
#define LL unsigned long long
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n,k,top;
LL a[MAXN],b[MAXN],ans1,ans2;
inline void in(LL &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
void dfs(int x,LL sum)
{
    if (x>top)
    {
        LL sum1=0,sum2=1;
        for (int i=1;i<=k;i++)  sum1*=sum,sum2*=sum,sum1+=(sum2>>top),sum2&=((1<<top)-1);
        ans1+=sum1;ans2+=sum2;ans1+=(ans2>>top);ans2&=((1<<top)-1);
        return;
    }
    dfs(x+1,sum);dfs(x+1,sum^a[x]);
}
int main()
{
    scanf("%d%d",&n,&k);for (int i=1;i<=n;i++)  in(a[i]);
    if (k==1)
    {
        for (int i=1;i<=n;i++)  ans1|=a[i];
        return printf(ans1&1?"%llu.5\n":"%llu\n",ans1>>1),0;
    }
    if (k==2)
    {
        top=1;
        for (int i=1;i<=n;i++)  b[1]|=a[i];
        for (int i=1;i<=n;i++)  for (int j=1;j<=top;++j)
        {
            if (!(b[j]&(~a[i])))   continue;
            if (!(b[j]&a[i]))   continue;
            b[++top]=b[j]&a[i];b[j]&=~a[i];
        }
        for (int i=1;i<=top;i++)    for (int j=i;j<=top;j++)    ans1+=b[i]*b[j];
        return printf(ans1&1?"%llu.5\n":"%llu\n",ans1>>1),0;
    }
    for (int i=(1<<22);i;i>>=1)
    {
        int j=top+1;
        for (;j<=n;j++) if (a[j]&i) break;
        if (j>n)    continue;swap(a[j],a[++top]);
        for (j=1;j<=n;j++)  if (j!=top&&(a[j]&i))   a[j]^=a[top];
    }
    dfs(1,0);printf(ans2?"%llu.5\n":"%llu\n",ans1);
}