某 SCOI 模拟赛 T1~T3【组合数学 分段打表 01-Trie Boruvka 树哈希 状压DP】

因为题目相比其他几次水一点所以就写一起了。

T1

题意

问有 1 到 $n$ 这 $n$ 个元素的二叉堆个数，模 $10^9+7$。$n\le 10^9$。

题解

记 $s{z}_i$ 为编号为 $i$ 的结点的子树的大小，知道 $s{z}_i$ 后我们很容易能够算出一个节点左右子树的大小。

以下有两种推导方式：

• 记 $f_i$ 为大小为 $i$ 的二叉堆的个数，于是 $f_1=f_2=1$，$f_{s{z}_x}=f_{s{z}_{lson}}\cdot f_{s{z}_{rson}}\cdot \left(\genfrac{}{}{0ex}{}{s{z}_x-1}{s{z}_{lson}}\right)$。再往下推亿点点可以得到答案为 $\frac{n!}{\prod _{i=1}^{n}s{z}_i}$；
• 先考虑给每个节点随便分配值，方案数为 $n!$；对于每个子树，我们要求只能是这个子树里面 $s{z}_i$ 个元素中最大的 1 个元素放到根，因此要除以 $s{z}_i$；最终方案数为 $\frac{n!}{\prod _{i=1}^{n}s{z}_i}$。

分子部分的阶乘显然不能现场算，于是分段打表。分母部分可以考虑 $g_{s{z}_x}=g_{s{z}_{lson}}\cdot g_{s{z}_{rson}}\cdot s{z}_x$，记忆化一下就是 $O(\log n)$。

总复杂度为 $O(\log n+\frac{n}{k})$。其中 $k$ 是分段打表的段数。

代码：

#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const int N=1e6+10,mod=1e9+7;
map<int,int>f;
map<int,int>ff;
int fac[N],ifac[N];
void init_fac(int n){
	fac[0]=1;
	for(int i=1;i<=n;i++)fac[i]=fac[i-1]*1ll*i%mod;
	ifac[0]=ifac[1]=1;
	for(int i=2;i<=n;i++)ifac[i]=ifac[mod%i]*1ll*(mod-mod/i)%mod;
	for(int i=1;i<=n;i++)ifac[i]=ifac[i-1]*1ll*ifac[i]%mod;
}
int getfac(int x){ return fac[x]; }
int getifac(int x){ return ifac[x]; }
int cnt=0;
int solve(int n){
	if(n<=2)return 1;
	map<int,int>::iterator it=f.find(n);
	if(it!=f.end())return it->second;
	++cnt;
	int t=1;
	while(t<<1<=n+1)t<<=1;--t;
	int l=min(t,n-(t>>1)-1),r=n-l-1;
	int c=getfac(l+r)*1ll*getifac(l)%mod*getifac(r)%mod;
	return f[n]=c*1ll*solve(l)%mod*solve(r)%mod;
}

int table[]={1,491101308,723816384,27368307,199888908,927880474,661224977,970055531,195888993,547665832,933245637,
368925948,136026497,135498044,419363534,668123525,30977140,309058615,189239124,940567523,429277690,
358655417,780072518,275105629,99199382,733333339,608823837,141827977,637939935,848924691,724464507,
326159309,903466878,769795511,606241871,957939114,852304035,375297772,624148346,624500515,203191898,
629786193,814362881,116667533,627655552,586445753,193781724,83868974,965785236,377329025,698611116},step=2e7;
int solvee(int n){
	if(n<=1)return 1;
	map<int,int>::iterator it=ff.find(n);
	if(it!=ff.end())return it->second;
	//cerr<<"> "<<n<<endl;
	int t=1;
	while(t<<1<=n+1)t<<=1;--t;
	int l=min(t,n-(t>>1)-1),r=n-l-1;
	return ff[n]=n*1ll*solvee(l)%mod*solvee(r)%mod;
}
int qpow(int x,int y){
	int ans=1;
	while(y){
		if(y&1)ans=ans*1ll*x%mod;
		x=x*1ll*x%mod;
		y>>=1;
	}
	return ans;
}
int main(){
	int n=getint();
	if(n<=1e6){
		init_fac(n);
		cout<<solve(n)<<endl;
		//cerr<<cnt<<endl;
		return 0;
	}
	int t=n/step;
	int ans=table[t];
	//cerr<<t<<" "<<step*t+1<<" "<<n<<endl;
	for(int i=step*t+1;i<=n;i++)ans=ans*1ll*i%mod;
	ans=ans*1ll*qpow(solvee(n),mod-2)%mod;
	cout<<ans<<endl;
	return 0;
}

T2

题意

有一个长为 $n$ 的未知序列 $a$，每个位置已知权值 $w_i$。你可以花费 $\underset{i=l}{\overset{r}{⨁}}{w}_i$ 的代价获得 $\underset{i=l}{\overset{r}{⨁}}{a}_i$ 的值。问：知道花费多少代价才能知道所有 $a_i$ 的值。$n\le 10^5$，$w_i\le 10^9$。

题解

记 $s_i$ 为 $a_i$ 的前缀异或和，记 $t_i$ 为 $w_i$ 的前缀异或和。特别的，$s_0=t_0=0$。

显然得到所有 $a_i$ 就是得到所有 $s_i$，而花费 $\underset{i=l}{\overset{r}{⨁}}{w}_i$ 的代价获得 $\underset{i=l}{\overset{r}{⨁}}{a}_i$ 的值就相当于花费 $t_r\oplus t_{l-1}$ 的代价获得 $s_r\oplus s_{l-1}$ 的值。

我们一开始只知道 $s_0=0$。

一次查询实际上可以看作在两点之间用给定代价连边；当所有点都与 $s_0$ 直接或间接连通时，我们就可以推出所有 $s_i$ 的值，也就是要求这个图的最小生成树。

于是这题就等价于 CF888G Xor-MST 了。

代码：

#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	} 
	return ans*f;
}
const int N=1e5+10,L=30,M=N*L;
int ch[M][2],sz[M],cnt=1;
void add(int x){
	int u=1;
	for(int i=L-1;i>=0;--i){
		int t=(x>>i)&1;
		sz[u]++;
		if(!ch[u][t])ch[u][t]=++cnt;
		u=ch[u][t];
	}
	sz[u]++;
}
int query(int x,int val,int l){
	int ans=0;
	for(int i=l;i>=0;--i){
		int t=(val>>i)&1;
		if(ch[x][t])x=ch[x][t];
		else x=ch[x][t^1],ans|=(1<<i);
	}
	return ans;
}
vector<int>v;
long long ans=0;
void solve(int u,int val,int l){
	//cerr<<"solve "<<u<<" "<<val<<" "<<l<<endl;
	if(l<0){
		for(int i=0;i<sz[u];i++)v.push_back(val);
		//cerr<<">> "<<val<<endl;
		return;
	}
	if(!ch[u][0]){
		solve(ch[u][1],val|(1<<l),l-1);
		return;
	}
	if(!ch[u][1]){
		solve(ch[u][0],val,l-1);
		return;
	}
	int c1=ch[u][0],c2=ch[u][1];
	if(sz[c1]>sz[c2])swap(c1,c2);
	solve(c2,val|((c2==ch[u][1])<<l),l-1);
	int qaq=v.size();
	solve(c1,val|((c1==ch[u][1])<<l),l-1);
	int mn=0x7f7f7f7f;
	for(int i=qaq;i<v.size();i++){
		mn=min(mn,(1<<l)|query(c2,v[i],l-1));
	}
	ans+=mn;
	return;
}

int a[N];
int main(){
	int n=getint();
	add(0);
	for(int i=1;i<=n;i++){
		a[i]=getint()^a[i-1];
		add(a[i]);
		//cerr<<"add "<<a[i]<<endl;
	}
	solve(1,0,L-1);
	cout<<ans;
	return 0;
}

T3

题意

有两棵树 $A$ 与 $B$，大小分别为 $n$ 与 $m$，问有多少 $A$ 的连通子图与 $B$ 同构，答案模 $10^9+7$。$n\le 2000$，$m\le 12$。

题解

一开始以为 $A$ 是一张图然后就没法做了。

首先判断两棵树同构的方式是：选定一个根，对于每个节点排序其所有子树的树哈希，然后再把这些树哈希按正常方式合并起来。

于是枚举 $B$ 中的每个点（以之为根时形态相同者不再重复枚举），对于每个有相同形态子树的节点，给它记录一个系数 $g[i]$ 为各种形态子树的数量的阶乘之积的倒数（相同形态的子树任意交换后都是重复的）。然后在 $A$ 上 DP，记录 $dp[i][j]$ 为 $A$ 中以 $i$ 为根，与 $B$ 中 $j$ 的子树同构的子图的个数，转移时枚举子树，状压一下。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const ll mod=1e9+7,G=19;
const int N=2010,M=14;
struct bian{
	int e,n;
};
struct graph{
	bian b[N<<1];
	int s[N],tot=0;
	void add(int x,int y){
		++tot;
		b[tot].e=y;
		b[tot].n=s[x];
		s[x]=tot;
	}
} A,B;
int m,n;

ll qpow(int x,int y){
	int ans=1;
	while(y){
		if(y&1)ans=ans*1ll*x%mod;
		x=x*1ll*x%mod;
		y>>=1;
	}
	return ans;
}
ll pw[N];
int ifac[N];

ll h[N];
int sz[N],g[N];
vector<int>ch[N];
bool cmp(int x,int y){ return h[x]<h[y]; }
void ss1(int x,int f){
	//cerr<<"ss1 "<<x<<" "<<f<<endl;
	ch[x].clear();h[x]=0;sz[x]=1;
	for(int i=B.s[x];i;i=B.b[i].n){
		if(B.b[i].e==f)continue;
		ss1(B.b[i].e,x);
		ch[x].push_back(B.b[i].e);
	}
	sort(ch[x].begin(),ch[x].end(),cmp);
	for(int i=0;i<ch[x].size();i++){
		int v=ch[x][i];
		//cerr<<x<<" -> "<<v<<endl;
		h[x]=h[x]*pw[sz[v]<<1]+h[v];
		sz[x]+=sz[v];
	}
	h[x]=h[x]*pw[sz[x]*2-1]+1;
	//cerr<<"h "<<x<<" "<<h[x]<<"|"<<sz[x]<<endl;
	g[x]=1;int l=0;
	for(int i=0;i<=ch[x].size();i++){
		if(i==ch[x].size()||h[ch[x][i]]!=h[ch[x][l]]){
			g[x]=g[x]*1ll*ifac[i-l]%mod;
			l=i;
		}
	}
}

int dp[N][M];
void ss(int x,int fa){
	for(int i=A.s[x];i;i=A.b[i].n){
		int v=A.b[i].e;
		if(v==fa)continue;
		ss(v,x);
	}
	for(int i=1;i<=m;i++){
		static int ic[M];
		int ct=0;
		for(int j=1;j<=m;j++)ic[j]=-1;
		for(int j=0;j<ch[i].size();j++)ic[ch[i][j]]=ct++;
		if(!ct){
			//cerr<<"!ct "<<i<<endl;
			dp[x][i]=1;continue;
		}
		static int f[2][1<<M];
		int S=1<<ct,t=0;
		for(int j=0;j<S;j++)f[0][j]=0;f[0][0]=1;
		for(int j=A.s[x];j;j=A.b[j].n){
			int v=A.b[j].e;
			if(v==fa)continue;
			for(int k=0;k<S;k++)f[t^1][k]=0;
			for(int k=1;k<=m;k++){
				if(~ic[k]){
					for(int l=(1<<ic[k]);l<S;l=(l+1)|(1<<ic[k])){
						f[t^1][l]+=f[t][l^(1<<ic[k])]*1ll*dp[v][k]%mod;
						if(f[t^1][l]>=mod)f[t^1][l]-=mod;
					}
				}
			}//for k
			t^=1;
			for(int k=0;k<S;k++){
				f[t][k]+=f[t^1][k];
				if(f[t][k]>=mod)f[t][k]-=mod;
			}
		}//for j
		dp[x][i]=g[i]*1ll*f[t][S-1]%mod;
		//cerr<<"dp "<<x<<" "<<i<<" "<<dp[x][i]<<" "<<g[i]<<" "<<f[t][S-1]<<endl;
	}//for i
}

int main(){
	n=getint();
	for(int i=1;i<n;i++){
		int x=getint(),y=getint();
		A.add(x,y); A.add(y,x);
	}
	m=getint();
	for(int i=1;i<m;i++){
		int x=getint(),y=getint();
		B.add(x,y); B.add(y,x);
	}
	
	pw[0]=1;for(int i=1;i<=m*2;i++)pw[i]=pw[i-1]*1ll*G;
	ifac[0]=ifac[1]=1;for(int i=2;i<=m;i++)ifac[i]=(ifac[mod%i]*1ll*(mod-mod/i))%mod;
	for(int i=1;i<=m;i++)ifac[i]=ifac[i-1]*1ll*ifac[i]%mod;
	
	ll ans=0;set<int>se;
	for(int i=1;i<=m;i++){
		ss1(i,0);
		//cerr<<">"<<i<<" "<<h[i]<<endl;
		if(!se.insert(h[i]).second)continue;
		ss(1,0);
		for(int j=1;j<=n;j++){
			ans+=dp[j][i];
			if(ans>=mod)ans-=mod;
		}
	}
	cout<<ans;
}