[cf1336E]Chiori and Doll Picking
作者：互联网
将所有$a_{i}$在二进制下展开，得到一个$n\times m$的01矩阵
对该矩阵做高斯消元（显然不影响结果），并要求得到如下的形式
$$
\left|\begin{array}{ll}
1&0&0&\cdots&0&\cdots\\
0&1&0&\cdots&0&\cdots\\
0&0&1&\cdots&0&\cdots\\
\vdots&\vdots&\vdots&\ddots&\vdots&\ddots\\
0&0&0&\cdots&1&\cdots
\end{array}\right|
$$
形式：假设其有$k$行，左边$k$列中仅主对角线为1（其余均为0），右边$m-k$列任意
另外，其余的$n-k$个$a_{i}$均为0，进而贡献即将答案乘上$2^{n-k}$（以下不再考虑）
在此基础上，暴力枚举$a_{i}$是否选择并$o(1)$计算$cnt(x)$（$x$在二进制下1的个数），时间复杂度为$o(2^{k})$
令$f_{i,j,x}$表示$a_{[1,i]}$中选$j$个且右侧结果为$x$的方案数（左边1的个数即为$j$），时间复杂度为$o(k^{2}2^{m-k})$
结合上述两个做法，对$k\le \frac{m}{2}+\log m$分类讨论，总复杂度为$o(m2^{\frac{m}{2}})$
对于$m\le 53$的hard version，考虑优化上述做法后半部分的复杂度：
记$A_{x}$为结果为$x$的方案数，注意到任意两种选法左侧均不同，即$A_{x}\in \{0,1\}$
构造$G^{c}_{x}=[cnt(x)=c]$，那么$ans_{c}=(A\bigoplus G^{c})_{0}$即为答案（其中$\bigoplus$为异或卷积）
使用FWT计算$\bigoplus$，结合其式子，不难得到（答案为）
$$
ans_{c}=\frac{\sum_{x=0}^{2^{m}-1}{\rm FWT}(A)_{x}\cdot{\rm FWT}(G^{c})_{x}}{2^{m}}
$$
性质1：$\forall 0\le x<2^{m},{\rm FWT}(A)_{x}\in \{0,2^{k}\}$
根据$A$的意义，不难得到$A_{x}=1\Longrightarrow \forall 0\le i<2^{m},A_{i}=A_{x\oplus i}$
考虑$A\bigoplus A$，代入可得$(A\bigoplus A)_{x}=\sum_{A_{i}=1}A_{x\oplus i}=\sum_{A_{i}=1}A_{x}=2^{k}A_{x}$
使用FWT计算$\bigoplus$，即$\big({\rm FWT}(A)_{x}\big)^{2}=2^{k}{\rm FWT}(A)_{x}$，解得${\rm FWT}(A)_{x}\in \{0,2^{k}\}$，即得证
另一方面，注意到${\rm FWT}(G^{c})_{x}$仅由$cnt(x)$确定，即
$$
{\rm FWT}(G^{c})_{x}=\sum_{i=0}^{c}(-1)^{i}{cnt(x)\choose i}{m-cnt(x)\choose c-i}
$$
枚举$cnt(x)$，结合性质1，仅需统计其中${\rm FWT}(A)_{x}=2^{k}$的$x$个数
记该值为$F_{cnt(x)}$，注意到其与$c$无关，同时求出$F$后代入之前的式子即可$o(m^{3})$求出$ans$
性质2：${\rm FWT}(A)_{x}=2^{k}$当且仅当$\forall 1\le j\le k,2\mid cnt(x\&a_{j})$
代入FWT的式子，即${\rm FWT}(A)_{x}=\sum_{A_{i}=1}(-1)^{cnt(x\&i)}\le 2^{k}$
同时，等号取到的条件为$\forall A_{i}=1,2\mid cnt(x\&i)$，显然$A_{a_{j}}=1$即得到必要性
关于充分性，注意到$cnt(i\&(x\oplus y))\equiv cnt(i\&x)+cnt(i\&y)(mod\ 2)$
对于$A_{i}=1$，存在集合$S$使得$\bigoplus_{j\in S}a_{j}=i$，进而$cnt(x\&i)\equiv \sum_{j\in S}(x\&a_{j})\equiv 0(mod\ 2)$
综上，即得证
枚举$x$的后$m-k$位$x'$，结合性质2，有$cnt(x)=cnt(x')+\sum_{j=1}^{k}[2\not\mid cnt(x'\&a_{j})]$
暴力枚举$x'$，并在过程中维护$\forall 1\le j\le k,[2\not\mid cnt(x'\&a_{j})]$，最终统计到$F_{cnt(x)}$即可
后半部分的复杂度被降为$o(2^{m-k}+m^{3})$，对$k\le \frac{m}{2}$分类讨论，总复杂度为$o(2^{\frac{m}{2}}+m^{3})$
结合初始的高斯消元，最终总复杂度为$o(nm+2^{\frac{m}{2}}+m^{3})$，可以通过
1 #include<bits/stdc++.h>
2 using namespace std;
3 #define N 200005
4 #define M 60
5 #define mod 998244353
6 #define ll long long
7 #define count __builtin_popcountll
8 int n,m,k,P[M];
9 ll a[N];
10 ll change(ll x){
11 ll ans=0;
12 for(int i=0;i<m;i++)
13 if ((x>>P[i])&1)ans|=(1LL<<i);
14 return ans;
15 }
16 namespace Subtask1{
17 int ans[M];
18 void dfs(int i,ll s){
19 if (i==k){
20 ans[count(s)]++;
21 return;
22 }
23 dfs(i+1,s),dfs(i+1,(s^a[i]));
24 }
25 void main(){
26 dfs(0,0);
27 int s=1;
28 for(int i=0;i<n-k;i++)s=(s<<1)%mod;
29 for(int i=0;i<=m;i++)ans[i]=(ll)ans[i]*s%mod;
30 for(int i=0;i<=m;i++)printf("%d ",ans[i]);
31 }
32 };
33 namespace Subtask2{
34 int C[M][M],F[M],ans[M];
35 ll b[M];
36 void dfs(int i,ll s){
37 if (i==m){
38 F[count(s)]++;
39 return;
40 }
41 dfs(i+1,s),dfs(i+1,(s^b[i]));
42 }
43 void main(){
44 for(int i=k;i<m;i++){
45 b[i]=(1LL<<i);
46 for(int j=0;j<k;j++)
47 if ((a[j]>>i)&1)b[i]|=(1LL<<j);
48 }
49 dfs(k,0);
50 for(int i=0;i<=m;i++){
51 C[i][0]=C[i][i]=1;
52 for(int j=1;j<i;j++)C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
53 }
54 for(int i=0;i<=m;i++)
55 for(int j=0;j<=m;j++){
56 int s=0;
57 for(int k=0;k<=i;k++){
58 int ss=(ll)C[j][k]*C[m-j][i-k]%mod;
59 if (k&1)ss=mod-ss;
60 s=(s+ss)%mod;
61 }
62 ans[i]=(ans[i]+(ll)s*F[j])%mod;
63 }
64 int s=1;
65 for(int i=0;i<n-m;i++)s=(s<<1)%mod;
66 for(int i=0;i<m-n;i++)s=(ll)s*(mod+1>>1)%mod;
67 for(int i=0;i<=m;i++)ans[i]=(ll)ans[i]*s%mod;
68 for(int i=0;i<=m;i++)printf("%d ",ans[i]);
69 }
70 };
71 int main(){
72 scanf("%d%d",&n,&m);
73 for(int i=0;i<n;i++)scanf("%lld",&a[i]);
74 for(int i=0;i<m;i++)P[i]=i;
75 for(int i=0;i<m;i++){
76 int pos=-1;
77 for(int j=k;j<n;j++)
78 if ((a[j]>>i)&1){
79 pos=j;
80 break;
81 }
82 if (pos<0)continue;
83 swap(P[k],P[i]),swap(a[k],a[pos]);
84 for(int j=0;j<n;j++)
85 if ((j!=k)&&((a[j]>>i)&1))a[j]^=a[k];
86 k++;
87 }
88 for(int i=0;i<k;i++)a[i]=change(a[i]);
89 if (k<=(m>>1))Subtask1::main();
90 else Subtask2::main();
91 return 0;
92 }
View Code
标签：cnt,le,复杂度,Chiori,cf1336E,cdots,FWT,rm,Picking 来源： https://www.cnblogs.com/PYWBKTDA/p/15942440.html