JZOJ5414. 【NOIP2017提高A组集训10.22】幸运值

Description

校庆志愿者小Z在休息时间和同学们玩卡牌游戏。一共有n张卡牌，每张卡牌上有一个数Ai，每次可以从中选出k张卡牌。一种选取方案的幸运值为这k张卡牌上数的异或和。小Z想知道所有选取方案的幸运值之和除以998244353的余数。

Input

输入的第一行有两个整数n和k。
第二行有n个整数，表示序列A。

Output

一个整数表示答案。

Sample Input

输入1：
3 2
1 2 3
输入2：
10 5
123 456 789 987 654 321 101 202 303 404

Sample Output

输出1：
6
输出2：
130776

Data Constraint

对于30%的数据满足，1<=n<=20
对于另30%的数据满足，1<=n<=100,0

Hint

样例1幸运值之和为(1 ⊕ 2) + (1 ⊕ 3) + (2 ⊕ 3) = 6

题解

这里是求异或和，在二进制上面的每一位对答案的贡献都是独立的。
所以，就分开成每一位来处理。

对于某一位，假设它有x个1。
在异或当中1只有是奇数个最后的异或结果才会为1。
枚举一个1的个数，奇数i
那么方案数就是选1的方案数*选0的方案数。
即$C_x^i*C_{n-x}^{k-i}$

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define ll long long
#define N 100003
#define db double
#define P putchar
#define G getchar
#define mo 998244353
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

ll jc[N],ny[N],ans,z[33];
int n,k,x,t[33];

ll ksm(ll x,int y)
{
    ll s=1;
    while(y)
    {
        if(y&1)s=s*x%mo;
        x=x*x%mo;
        y>>=1;
    }
    return s;
}

ll C(int x,int y)
{
    if(x>y)return 0;
    return jc[y]*ny[x]%mo*ny[y-x]%mo;
}

int main()
{
    freopen("card.in","r",stdin);
    freopen("card.out","w",stdout);
    z[0]=1;
    for(int i=1;i<32;i++)
        z[i]=z[i-1]*2;
    read(n);read(k);
    ny[0]=jc[0]=1;
    for(int i=1;i<=n;i++)
        jc[i]=jc[i-1]*i%mo,ny[i]=ksm(jc[i],mo-2);
    for(int i=1;i<=n;i++)
    {
        read(x);
        for(int j=0;j<32;j++)
            if((x&z[j])>0)t[j]++;
    }

    for(int i=0;i<32;i++)
        for(int j=1;j<=min(t[i],k);j=j+2)
            ans=(ans+C(j,t[i])*C(k-j,n-t[i])%mo*z[i]%mo)%mo;
    write(ans);
}