SSL2795 2017年10月26日提高组T2 幸运值（组合）

2017年10月26日提高组T2 幸运值

Description

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

Input

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

Output

一个整数表示答案。

分析：我们发现每一位的答案与选了哪些数无关，而与哪一位的0、1数量有关。

于是我们统计出这 N 个数的每一位的0、1数。

对于每一位，我们设有 x 个 1 ，那么则有 n−x 个 0 。

要在其中选 k 个数，又发现其异或和只与 1 的个数的奇偶性有关。

那么这一位的答案即为：

∑ C(j,x)∗C(k−j,n−x)(1<=j<=x,j mod 2=1)
即选一些数作为1、另一些数作为0的组合数。

代码

#include <cstdio>
#define N 100005
#define mo 998244353
typedef long long ll;
using namespace std;

ll jc[N],ny[N],f[N];
int n,x;

ll power(ll a,ll b)
{
    ll s=1;
    while (b)
    {
        if (b&1) s=s*a%mo;
        a=a*a%mo;
        b>>=1;
    }
    return s;
}

ll min(ll x,ll y)
{
    return x<y?x:y;
}

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

int main()
{
    freopen("card.in","r",stdin);
    freopen("card.out","w",stdout);
    int k;
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        int p=0;
        while (x>0)
        {
            f[p++]+=x&1;
            x>>=1;
        }
    }
    jc[0]=ny[0]=1;
    for (int i=1;i<=n;i++)
    {
        jc[i]=jc[i-1]*i%mo;
        ny[i]=power(jc[i],mo-2);
    }
    ll ans=0;
    for (int i=0;i<30;i++)
    {
        ll s=0;
        for (int j=1;j<=min(k,f[i]);j+=2)
            s=(s+C(f[i],j)*C(n-f[i],k-j)%mo)%mo;
        ans=(ans+s*power(2,i)%mo)%mo;
    }
    printf("%lld",ans);
}