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<Ai<=1024
对于80%的数据满足,1<=n<=2000
对于100%的数据满足,1<=n<=100000,0<Ai<2^31,1<=k<=n
Solution
比赛时我只打了一个暴力,不知道为什么还打错了。
但是这个正解很好理解:
首先我们知道因为题目涉及到的异或运算时关于二进制的运算,所以我们很容易想到可以把每个数分别拆成二进制形式。因为位运算只涉及到每一位,所以直接做。
对于每一位,我们统计这些数在这一位上含有一的个数,设其为x,我们得到一个公式:
∑i=1kC(x,i)∗C(n−x,k−i)∗2j
我们来理解一下这条式子。
首先,i必须是一个奇数,因为只有有奇数个1时才可以是这一位算出来的值不为零,那么两个排列组合就很简单了,那个 2j 表示当前这一位为j,就是这一位的贡献。
然后我们的排列组合因为要在模意义下,所以要用到逆元。
#include<cstdio>
#include<iostream>
#define p 998244353
using namespace std;
int n,k;
long long ans;
long long fac[100001];
int a[100001];
long long ksm(long long xx,long long k)
{
long long y=xx,t=k,s=1;
while(t)
{
if(t&1)(s*=y)%=p;
y=(y*y)%p;
t/=2;
}
return s;
}
int q()
{
int x=0;
for (int i=1;i<=n;++i)
{
if(a[i]&1)++x;
a[i]/=2;
}
return x;
}
long long C(int n,int m)
{
if(n==m||m==0)return 1;
if(n<m) return 0;
return ((fac[n]*ksm(fac[m],p-2))%p*ksm(fac[n-m],p-2))%p;
}
int main()
{
freopen("card.in","r",stdin);
freopen("card.out","w",stdout);
scanf("%d%d",&n,&k);
int i,j=0,x;fac[1]=1;
for (i=1;i<=n;++i) scanf("%d",&a[i]);
for (i=2;i<=n;++i) fac[i]=(fac[i-1]*i)%p;
while (x=q())
{
for (i=1;i<=k&&i<=x;i+=2) (ans+=(C(x,i)*C(n-x,k-i))%p*ksm(2,j))%=p;
++j;
}
printf("%lld\n",ans);
}