CF786E ALT

题意

有一棵 \(n\) 个点的树和 \(m\) 个人,第 \(i\) 个人从 \(u_i\) 走到 \(v_i\)

现在要发宠物,要求一个人要么他自己发到宠物,要么他走的路径上的都有宠物。

求最小代价,并输出任意方案。

\(n,m \leq 20000\)
传送门

思路

对每个人和每条树边都建一个点。

源点向每个人连容量 \(1\) 的边,每条树边向汇点连容量 \(1\) 的边。每个人向他要走到的所有边连容量 \(+\infty\) 的边。

给人发就是割掉人与源的边,放边上就是割树边与汇的边,人与树边间的边不能割。当源汇不连通的时候,就是满足题意的,题目转化为最小割。

问题出在连边上,这是 \(nm\) 的,考虑如何优化连边。线段树?树上问题有点麻烦,想到倍增。

对于一段相邻的\(2^i\)个点,都建一个点与它们所有相连,其实又有点像线段树,然后就会对于每条链得到一棵类似线段树的东西?大区间连向小区间,看起来就是。

然后对每个人,将他的路径在 LCA 处分成两条路径,这两条路径分别向对应区间覆盖。
那么我们得到的新图有 \(n\log n+m\) 个点和 \(n\log n+4m\) 条边,可以通过。

这里还要输出方案。

我们知道,最小割中的边一定满流。(全局最大流,分成两个集合,其间的边一定是满流的,而也就是最小割)因此我们从源点开始 dfs,只走没满流的边,并标记被 dfs 到的点。则图被分成两部分,一部分被访问过,一部分没被访问过。其中间那些边就是一个最小割。

然后,如果一个人代表的点没被访问过,则说明他所属那条边被割了。如果一条树边代表的点被访问过,则说明它所属那条边被割了。记录标号输出

#include <bits/stdc++.h>
using std::queue;
const int W=14,N=20005*(W+1),M=N+N*4;
int to[M<<1],w[M<<1],Next[M<<1],edge,n,x,y,f[20005][W+1],idn[N],deep[N],last[N],b[20005][W+1];
int s,t,m,cnt,ans,tag[N],cur[N];
queue <int> q;
void add(int x,int y,int z){
	to[++edge]=y;
	Next[edge]=last[x];
	last[x]=edge;
	w[edge]=z;
}
void dfs(int x,int fa){
	f[x][0]=fa,deep[x]=deep[fa]+1;
	for (int i=last[x];i;i=Next[i])
		if (to[i]!=fa) {
			idn[to[i]]=(i+1)/2;
			dfs(to[i],x);
		}
}
void Add(int x,int y,int w){
	add(x,y,w);
	add(y,x,0);
}
int lca(int x,int y){
	if (deep[x]<deep[y]) std::swap(x,y);
	for (int i=W;i>=0;i--)
		if (deep[f[x][i]]>=deep[y]) x=f[x][i];
	if (x==y) return x;
	for (int i=W;i>=0;i--)
		if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}
void addedge(int x,int y,int id){
	for (int i=W;i>=0;i--)
		if (deep[f[x][i]]>=deep[y]){
			Add(id,b[x][i],n+1);
			x=f[x][i];
		}
}
bool bfs(){
	for (int i=0;i<=t;i++) cur[i]=last[i],deep[i]=0;
	deep[s]=1;
	q.push(s);
	while (!q.empty()){
		int x=q.front();
		q.pop();
		for (int i=last[x];i;i=Next[i])
			if (w[i] && !deep[to[i]]){
				deep[to[i]]=deep[x]+1;
				q.push(to[i]);
			}
	}
	return (deep[t]);
}
int c(int x) { return x&1?x+1:x-1; }
int dfs(int x){
	if (x==t) return 1;
	for (int i=cur[x];i;i=Next[i]){
		cur[x]=i;
		int u=to[i];
		if (deep[u]>deep[x] && w[i]){
			int di=dfs(to[i]);
			if (di){
				w[i]--;
				w[c(i)]++;
				return 1;
			}
		}
	}
	return 0;
}
void DFS(int x){
	tag[x]=1;
	for (int i=last[x];i;i=Next[i])
		if (w[i] && !tag[to[i]]) DFS(to[i]);
}
int main(){
	scanf("%d%d",&n,&m);
	for (int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y,0),add(y,x,0);
	}
	s=0;
	dfs(1,0);
	edge=0;memset(last,0,sizeof(last));
	cnt=n;
	for (int i=2;i<=n;i++) b[i][0]=i;
	for (int i=1;i<=W;i++)
		for (int j=1;j<=n;j++)
			if (f[f[j][i-1]][i-1]){ 
				f[j][i]=f[f[j][i-1]][i-1];
				Add(++cnt,b[j][i-1],n+1);
				Add(cnt,b[f[j][i-1]][i-1],n+1);
				b[j][i]=cnt;
			} 
	for (int i=1;i<=m;i++){
		scanf("%d%d",&x,&y);
		Add(s,++cnt,1);
		idn[cnt]=i;
		int l=lca(x,y);
		addedge(x,l,cnt),addedge(y,l,cnt);
	}
	t=cnt+1;
	for (int i=2;i<=n;i++) Add(i,t,1);
	while (bfs())
		while (dfs(s)==1) ans++;
	printf("%d\n",ans);
	DFS(s);
	ans=0;
	for (int i=last[s];i;i=Next[i])
		if (tag[to[i]]!=tag[s]) ans++;
	printf("%d ",ans);
	for (int i=last[s];i;i=Next[i])
		if (tag[to[i]]!=tag[s]) printf("%d ",idn[to[i]]);
	puts("");
	ans=0;
	for (int i=last[t];i;i=Next[i])
	if (tag[to[i]]!=tag[t]) ans++;
	printf("%d ",ans);
	for (int i=last[t];i;i=Next[i])
		if (tag[to[i]]!=tag[t]) printf("%d ",idn[to[i]]);
	return 0;
} 

后记

追随神仙的脚步
抄袭于此Mrsrz

posted @ 2019-12-09 10:48  flyfeather  阅读(273)  评论(0编辑  收藏  举报