【学习笔记】NOI 模拟赛 t3 exam

仰望星空的蚂蚁

已于 2023-06-06 22:28:12 修改

阅读量540

点赞数 1

文章标签：学习笔记

于 2023-06-06 20:16:45 首次发布

本文链接：https://blog.csdn.net/cqbzlydd/article/details/131074695

版权

点这里看题

~~因为版权问题啥的也不好公开，就将就着看吧。~~

好题。考场上观察出了一点苗头，可惜没能 $A C$ 。

事实上，这道题目并不需要高深的数学知识，只不过需要用 $FMT$ 进行反复正变换和逆变换罢了。给出题人点赞。

首先这是一个或运算的卷积。考虑将或运算转化为子集和运算，因为我们对 $FMT$ 是熟知的。换句话说，对于 $i\in [0,2^m)$ ，记 $v_i=\prod_{j\le n}p_{j,i}$ ， ${p_j\}$ 显然就是做完子集和后的序列，长度为 $2^m$ ，但是 其本质不同的值只有 $2^k$ 个。最后对 $v_i$ 做逆 $FMT$ 就是答案，如果对 $FMT$ 比较熟知的话应该是可以快速反应过来的。~~我考场上就没这么快的反应能力。~~

观察上述问题。点乘是容易的，肯定不会 $O(n2^m)$ 暴力去乘，因为如果是有位运算参与的点乘的话可以先打上标记到最后再来做子集乘。显然我们也不能对于 $n$ 行都去做长度为 $2^m$ 的卷积，否则点乘也只能暴力了。

考虑替代的思想。先抛出结论，对第 $i$ 行的所有元素做长度为 $2^k$ 的逆 $FMT$ ，然后对于 $i\in [0,2^k)$ ，设这些数与起来的结果是 $set_{i}$ ，做完逆 $FMT$ 后结果是 $val_i$ ，那么 $v_{set_i}\gets v_{set_i}\times val_i$ 。只需要最后对长度为 $2^m$ 的数组做一次 $FMT$ 即可。

为什么这样是对的？不妨这样来看，对于一个特定的 $j\in [0,2^m)$ ，设有用的元素的集合为 $S_j$ ，显然应该点乘上 $val[S_j]$ 。显然 $S_j$ 的子集的 $set_v(v\sube S_j)$ 也是 $set_{S_j}$ 的子集，因为已经做了一遍逆 $FMT$ ，所以做完前缀积过后就是 $val[S_j]$ 。同时，如果 $v\nsubseteq S_j$ 那么也不会被统计到。于是就做完了。

复杂度 $O(m2^m+nk2^k)$ 。注意做逆 $FMT$ 的时候要乘的逆元其实是可以同时处理的，这样会少一个 $\log$ 。

自己做的时候yy了一些结论。最后发现还是没有上面写的简洁。看来还是在工具的选择上出了一点问题。

#include<bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
using namespace std;
const int mod=19260817;
//dj yyds!!!
int n,m,K,a[100005][7],sets[1<<7],ones[1<<7],vones[1<<22];
ll fpow(ll x,ll y=mod-2){
	ll z(1);
	for(;y;y>>=1){
		if(y&1)z=z*x%mod;
		x=x*x%mod;
	}
	return z;
}
ll p[100005][7],pp[1<<7],invpp[1<<7],v[1<<22],inv=fpow(1e6),res;
int lowbit(int x){return x&-x;}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
	cin>>n>>m>>K;
	for(int i=0;i<1<<m;i++)v[i]=1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<K;j++){
			cin>>p[i][j];
		}
		for(int j=0;j<K;j++){
			cin>>a[i][j];
		}
		for(int j=0;j<1<<K;j++){
			pp[j]=sets[j]=0;
			for(int k=0;k<K;k++){
				if(j>>k&1)pp[j]+=p[i][k],sets[j]|=a[i][k];
			}
			if(pp[j]!=0){
				invpp[j]=fpow(pp[j]),ones[j]=1;
			}
			else{
				pp[j]=invpp[j]=1,ones[j]=0;
			}
		}
		for(int j=0;j<K;j++){
			for(int k=0;k<1<<K;k++){
				if(k>>j&1){
					pp[k]=pp[k]*invpp[k^(1<<j)]%mod,ones[k]-=ones[k^(1<<j)];
					invpp[k]=invpp[k]*pp[k^(1<<j)]%mod;
				}
			}
		}
		for(int j=0;j<1<<K;j++){
			v[sets[j]]=v[sets[j]]*pp[j]%mod;
			vones[sets[j]]+=ones[j];
		}
	}
	for(int i=0;i<m;i++){
		for(int j=0;j<1<<m;j++){
			if(j>>i&1){
				v[j]=v[j]*v[j^(1<<i)]%mod,vones[j]+=vones[j^(1<<i)];
			}
		}
	}
	ll fac3=1,tmp=fpow(inv,n);
	for(int i=0;i<1<<m;i++){
		assert(vones[i]<=n);
		assert(vones[i]>=0);
		if(vones[i]!=n)v[i]=0;
	}
	for(int i=0;i<m;i++){
		for(int j=0;j<1<<m;j++){
			if(j>>i&1){
				v[j]=(v[j]-v[j^(1<<i)])%mod;
			}
		}
	}
	for(int i=0;i<1<<m;i++){
		v[i]=(v[i]*tmp%mod+mod)%mod;
		res^=(fac3*v[i]%mod);
		fac3=fac3*3%mod;
	}
	cout<<res;
}