IOI2020集训队作业-19 (CF571D, CF708E, ARC093F)

A - CF571D Campus

Sol

首先对大学和警卫室分别建个Kruskal重构树并求出dfs序。询问等价于：询问某个宿舍在最后一次被清空了之后，所有对它的加操作的和。可以差分一下转化成最后一次清空和询问这两个时间点之前的加操作的和，然后对时间做扫描线，用线段树维护加操作即可。

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#define PB push_back
#define ll long long
using namespace std;
template <class T>
inline void rd(T &x) {
	x=0; char c=getchar(); int f=1;
	while(!isdigit(c)) { if(c=='-') f=-1; c=getchar(); }
	while(isdigit(c)) x=x*10-'0'+c,c=getchar(); x*=f;
}
template <class T> inline void cmax(T &x,T y) { if(x<y) x=y; }
template <class T> inline void cmin(T &x,T y) { if(y<x) x=y; }
const int N=5e5+10,M=N<<1;
int n,m;
struct Tree {
	int F[M],fa[M],ch[M][2],ti[M],sz[M],ncnt;
	int dfn[M],lb[M],rb[M],id;
	void init() {
		ncnt=n;
		for(int i=1;i<=n;++i) fa[i]=i,sz[i]=1;
	}
	int find(int x) { return fa[x]==x?x:fa[x]=find(fa[x]); }
	void un(int x,int y,int t) {
		int fx=find(x),fy=find(y),u=++ncnt;
		fa[u]=fa[fx]=fa[fy]=u,ti[u]=t,sz[u]=sz[fx]+sz[fy];
		ch[u][0]=fx,ch[u][1]=fy;
	}
	void dfs(int u) {
		if(!ch[u][0]) {
			lb[u]=rb[u]=dfn[u]=++id;
		}
		else {
			lb[u]=id+1;
			dfs(ch[u][0]),dfs(ch[u][1]);
			F[ch[u][0]]=F[ch[u][1]]=u;
			rb[u]=id;
		}
	}
	void build() {
		for(int i=1;i<=ncnt;++i) if(find(i)==i) dfs(i);
		for(int i=1;i<=n;++i) fa[i]=i;
	}
	int get(int x,int t) {
		while(F[fa[x]]&&ti[F[fa[x]]]<=t) fa[x]=F[fa[x]];
		return fa[x];
	}
}A,B;

namespace seg {
	int cov[N<<2];
	inline void push_down(int c) {
		if(cov[c]) {
			cmax(cov[c<<1],cov[c]);
			cmax(cov[c<<1|1],cov[c]);
			cov[c]=0;
		}
	}
	void upd(int l,int r,int c,int L,int R,int v) {
		if(L<=l&&R>=r) return (void)(cov[c]=v);
		int mid=l+r>>1; push_down(c);
		if(L<=mid) upd(l,mid,c<<1,L,R,v);
		if(R>mid) upd(mid+1,r,c<<1|1,L,R,v);
	}
	int query(int l,int r,int c,int v) {
		if(l==r) return cov[c];
		int mid=l+r>>1; push_down(c);
		if(v<=mid) return query(l,mid,c<<1,v);
		else return query(mid+1,r,c<<1|1,v);
	}
}

namespace Bit {
	ll c[N];
	void add(int i,int t) { for(;i<=n;i+=i&-i) c[i]+=t; }
	void add(int l,int r,int t) { add(l,t),add(r+1,-t); }
	ll query(int i) { ll ans=0; for(;i;i-=i&-i) ans+=c[i]; return ans; }
}

struct OPT {
	int ty,t,x;
	OPT(int ty=0,int t=0,int x=0): ty(ty),t(t),x(x) {}
};
vector<OPT> Q;
ll ans[N];
struct Que {
	int p,id,sgn;
	Que(int p=0,int id=0,int sgn=0): p(p),id(id),sgn(sgn) {}
};
vector<Que> R[N];
char ty[20];
int main() {
	rd(n),rd(m); A.init(),B.init();
	for(int i=1,x,y;i<=m;++i) {
		scanf("%s",ty);
		if(ty[0]=='U') {
			rd(x),rd(y);
			A.un(x,y,i);
		}
		else if(ty[0]=='M') {
			rd(x),rd(y);
			B.un(x,y,i);
		}
		else {
			rd(x);
			Q.PB(OPT(ty[0]=='A'?1:(ty[0]=='Z'?2:3),i,x));
		}
	}
	A.build(),B.build();
	int qtot=0;
	for(int i=0;i<Q.size();++i)
		if(Q[i].ty==2) {
			int x=B.get(Q[i].x,Q[i].t);
			seg::upd(1,n,1,B.lb[x],B.rb[x],Q[i].t);
		}
		else if(Q[i].ty==3) {
			R[Q[i].t].PB(Que(A.dfn[Q[i].x],++qtot,1));
			int pre=seg::query(1,n,1,B.dfn[Q[i].x]);
			if(pre) R[pre].PB(Que(A.dfn[Q[i].x],qtot,-1));
		}
	int q=0;
	for(int i=1;i<=m;++i) {
		while(q<Q.size()&&Q[q].t<=i) {
			if(Q[q].ty==1) {
				int anc=A.get(Q[q].x,Q[q].t);
				Bit::add(A.lb[anc],A.rb[anc],A.sz[anc]);
			}
			q++;
		}
		for(int j=0;j<R[i].size();++j)
			ans[R[i][j].id]+=R[i][j].sgn*Bit::query(R[i][j].p);
	}
	for(int i=1;i<=qtot;++i) printf("%lld\n",ans[i]);
	return 0;
}fr

---

B - CF708E Student’s Camp

Sol

预处理出$k$天之后某一侧掉了$i$块砖的概率$p_i$。

一种暴力$dp$：设$f_{i,l,r}$表示考虑到了第$i$行，第$i$行最后剩下的砖是区间$[l,r]$，前$i$行都连通的概率。则$f_{i+1,l,r}=p_{l-1}\cdot p_{n-r}\cdot {\sum }_{[l,r]\cap [l^{\prime },r^{\prime }]\ne \varnothing }{f}_{i,l^{\prime },r^{\prime }}$。

考虑将转移中的${\sum }_{[l,r]\cap [l^{\prime },r^{\prime }]\ne \varnothing }{f}_{i,l^{\prime },r^{\prime }}$更改为$\sum f_{i,l^{\prime },r^{\prime }}-{\sum }_{r^{\prime }<l}{f}_{i,l^{\prime },r^{\prime }}-{\sum }_{l^{\prime }>r}{f}_{i,l^{\prime },r^{\prime }}$，则我们只需要知道$f{l}_{i,x}={\sum }_r{f}_{i,x,r}$和$f{r}_{i,x}={\sum }_l{f}_{i,l,x}$就可以更新了。

而另一方面，考虑如何直接由$f{l}_i,f{r}_i$得到$f{l}_{i+1},f{r}_{i+1}$：

$$f{r}_{i+1,x}=\sum _l{f}_{i+1,l,x}=\sum _l{p}_{l-1}\cdot p_{n-x}\cdot (\sum _{y\in [1,n]}f{r}_{i,y}-\sum _{y>x}f{l}_{i,y}-\sum _{y<l}f{r}_{i,y})$$

这个可以前缀和优化，复杂度$O(n^2)$。

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
template <class T>
inline void rd(T &x) {
	x=0; char c=getchar(); int f=1;
	while(!isdigit(c)) { if(c=='-') f=-1; c=getchar(); }
	while(isdigit(c)) x=x*10-'0'+c,c=getchar(); x*=f;
}
const int mod=1e9+7,N=1510;
int Pow(int x,int y) {
	int res=1;
	while(y) {
		if(y&1) res=res*(ll)x%mod;
		x=x*(ll)x%mod,y>>=1;
	}
	return res;
}
int fac[100010],inv[100010],p;
int P[N];
void getfac(int n) {
	fac[0]=1; for(int i=1;i<=n;++i) fac[i]=fac[i-1]*(ll)i%mod;
	inv[n]=Pow(fac[n],mod-2); for(int i=n;i>=1;--i) inv[i-1]=inv[i]*(ll)i%mod;
}
int n,m,L;
int C(int n,int m) { return fac[n]*(ll)inv[m]%mod*inv[n-m]%mod; }
int fr[N][N],sfr[N][N],fl[N][N],sfl[N][N];
int s2[N],s1[N];
int main() {
	rd(n),rd(m);
	{ int a,b; rd(a),rd(b); p=a*(ll)Pow(b,mod-2)%mod; }
	rd(L),getfac(L);
	for(int i=0;i<=min(L,m);++i) P[i]=C(L,i)*(ll)Pow((mod+1-p)%mod,L-i)%mod*Pow(p,i)%mod;

	sfr[0][m]=fr[0][m]=sfl[0][1]=fl[0][1]=1;
	
	for(int i=1;i<=m;++i) s1[i]=(s1[i-1]+P[i-1])%mod;
	
	for(int i=1;i<=n;++i) {
		for(int j=1;j<=m;++j) s2[j]=(s2[j-1]+P[j-1]*(ll)sfr[i-1][j-1])%mod;
		int tot=sfr[i-1][m];
		for(int j=1;j<=m;++j)
			fr[i][j]=P[m-j]*(ll)(s1[j]*(ll)(tot-sfl[i-1][j+1])%mod-s2[j])%mod;
		for(int j=1;j<=m;++j) {
			sfr[i][j]=(sfr[i][j-1]+fr[i][j])%mod;
			fl[i][j]=fr[i][m-j+1];
		}
		for(int j=m;j>=1;--j)
			sfl[i][j]=(sfl[i][j+1]+fl[i][j])%mod;
	}
	printf("%d",(sfr[n][m]+mod)%mod);
	return 0;
}

---

C - ARC093F Dark Horse

Sol

整个过程会形成一个完全二叉树的结构，每个叶子节点表示一个选手。

对于不包含$1$号节点的子树，最后的胜者是其中编号最小的选手，子树内节点的排列顺序对其结果没有影响。

考虑$1$向根走的过程中，会分别挑战大小为$2^0,2^1,2^2,\cdots 2^{n-1}$的子树中的胜者，而$1$获胜的条件是这些胜者都不是$a_1,a_2,\cdots a_m$。

考虑容斥，枚举 { a 1 , a 2 , ⋯ a m } \{a_1,a_2,\cdots a_m \} {a1​,a2​,⋯am​}的一个子集$S$作为最终的胜者，再枚举它们所在的子树大小。则要求对于每一个$a_x\in S$，它所在的子树内没有元素比它小。所以可以按照$a_x$从大到小分配与它在同一个子树内的元素，方案数就是$\left(\genfrac{}{}{0ex}{}{2^n-a_x-pre}{s{z}_x-1}\right)$，其中$pre$表示大于$a_x$的那些$a_y\in S$的$sz$的和，$s{z}_x$表示$a_x$所在的子树的大小。

将$a$降序排序，令$f[i][S]$表示考虑了前$i$个$a$，选择过的子树大小的和为$S$，所有的方案数。由于子树大小都是$2^i$，所以$S$同时也可以代表哪些子树大小是选择过的。枚举$a_i$对应的没有选过的子树大小就可以转移。

这样算出来的方案数是没有考虑$1$所在的位置的，所以最后还要乘以$2^n$。

时间复杂度$O(2^{n}nm)$。

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
template <class T>
inline void rd(T &x) {
	x=0; char c=getchar(); int f=1;
	while(!isdigit(c)) { if(c=='-') f=-1; c=getchar(); }
	while(isdigit(c)) x=x*10-'0'+c,c=getchar(); x*=f;
}
const int mod=1e9+7;
int Pow(int x,int y) {
	int res=1;
	while(y) {
		if(y&1) res=res*(ll)x%mod;
		x=x*(ll)x%mod,y>>=1;
	}
	return res;
}
inline void Add(int &x,int y) { x+=y; if(x>=mod) x-=mod; }
inline void Dec(int &x,int y) { x-=y; if(x<0) x+=mod; }
int f[1<<16],g[1<<16],sz[1<<16];
int A[20],m,n;
int N;
int fac[(1<<16)+10],inv[(1<<16)+10];
void getfac(int n) {
	fac[0]=1; for(int i=1;i<=n;++i) fac[i]=fac[i-1]*(ll)i%mod;
	inv[n]=Pow(fac[n],mod-2); for(int i=n;i>=1;--i) inv[i-1]=inv[i]*(ll)i%mod;
}
int C(int n,int m) { return fac[n]*(ll)inv[m]%mod*inv[n-m]%mod; }
int main() {
	rd(n),rd(m); N=1<<n,getfac(N);
	for(int i=1;i<=m;++i) rd(A[i]);
	f[0]=1;
	for(int i=m;i>=1;--i)
		for(int s=N-1;s>=0;--s) if(f[s]) {
			for(int j=0;j<n;++j) if(!(s>>j&1)&&N-A[i]+1>=s+(1<<j))
				Add(f[s|1<<j],f[s]*(ll)C(N-A[i]-s,(1<<j)-1)%mod);
		}
	int ans=0;
	for(int i=1;i<N;++i) sz[i]=sz[i&(i-1)]+1;
	for(int x=0;x<N;++x) {
		int tot=N-1-x;
		for(int j=n-1;j>=0;--j)
			if(tot>=(1<<j)) f[x]=f[x]*(ll)C(tot,1<<j)%mod,tot-=(1<<j);
		if(sz[x]&1) Dec(ans,f[x]);
		else Add(ans,f[x]);
	}
	for(int i=1;i<N;i<<=1) ans=ans*(ll)fac[i]%mod;
	for(int i=1;i<=n;++i) ans=ans*2ll%mod;
	printf("%d",ans);
	return 0;
}