agc029F Construction of a tree

agc029F Construction of a tree

• 给$n-1$个只包括$1-n$正整数的点集，要求从每一个点集中选择一对点连边，使得最后构成一个$n$个点的树。

• $n\le 1e5,\sum |S|\le 2e5$

Solution

• 很容易想到一个无解的条件：假设读入的点集是$S$，对于一个 T = { S } T=\{S\} T={S}，如果$T$涵盖的点数$c\le |T|$那么一定是无解的。因为这样连不少于点数条边一定会有环。
• 大胆猜测这就是充分条件。
• 考虑构造。
• 将集合看作一个点，想它包括的点连边，显然这是一个二分图。
• 考虑我们对于每一个集合连的点要选两个，我们不妨对于每个集合先选择一个，即找到一个最大匹配。
• 不妨设$n$个点在左部，$n-1$个点集代表的点在右侧。
• 如果这能构成一棵树，一定有$n-1$的最大匹配。
• 那么还有一个没有匹配到的点，我们从这个点开始找到这棵树。从当前已经达到的点集中尽可能地往外扩展，如果选择右部的点就直接选择它已经匹配的点，并新增一条边连接当前点和这个右部点匹配的点。
• 如果没有解，一定对应左侧当前遍历到的点与右侧剩下没有访问过的点集没有连边。那么右边和左边都剩下相同的点数，这样一定是无解的。
• 否则满足没有开头的必要条件，我们构造出了一个解。
• dinic求二分图匹配是$O(m\sqrt{n})$的，因为可以证明每一次增广的长度比前一次长。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 200005
#define maxm 1000005
using namespace std;

int n,i,j,k;
int S,T,em,e[maxm],nx[maxm],ls[maxn],cur[maxn],ec[maxm];

void read(int &x){
	x=0; char ch=getchar();
	for(;ch<'0'||ch>'9';ch=getchar());
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
}

void insert(int x,int y,int z){
	em++; e[em]=y; nx[em]=ls[x]; ls[x]=em; ec[em]=z;
	em++; e[em]=x; nx[em]=ls[y]; ls[y]=em; ec[em]=0;
}

int d[maxn],dis[maxn];
int BFS(){
	memset(dis,0,sizeof(int)*(T+1));
	int t=0,w=1; d[1]=S,dis[S]=1;
	while (t<=w){
		int x=d[++t]; cur[x]=ls[x];
		for(int i=ls[x];i;i=nx[i]) if (ec[i]&&!dis[e[i]])
			dis[e[i]]=dis[x]+1,d[++w]=e[i];
	}
	return dis[T]>0;
}

int dfs(int x,int p){
	if (x==T) return p;
	int res=p;
	for(int &i=cur[x];i;i=nx[i]) if (ec[i]&&dis[x]+1==dis[e[i]]){
		int tmp=dfs(e[i],min(res,ec[i]));
		res-=tmp,ec[i]-=tmp,ec[i^1]+=tmp;
		if (!res) break;
	}
	return p-res;
}

int bz[maxn],to[maxn],cnt,vis[maxn],ans[maxn][2];

void build(int x){
	cnt++;
	for(int i=ls[x];i;i=nx[i]) if (e[i]<S&&!vis[e[i]]){
		ans[e[i]-n][0]=x,ans[e[i]-n][1]=to[e[i]];
		vis[e[i]]=1,build(to[e[i]]);
	}
}

void maxflow(){
	int sum=0;
	while (BFS())
		sum+=dfs(S,2e9);
	if (sum!=n-1) printf("-1"),exit(0);
	for(int x=1;x<=n;x++) for(int i=ls[x];i;i=nx[i]) if (!ec[i]&&e[i]<S)
		bz[x]=e[i],to[e[i]]=x;
	int rt=0;
	for(int x=1;x<=n;x++) if (!bz[x]) rt=x;
	build(rt);
}

int main(){
	freopen("ceshi.in","r",stdin);
	read(n),em=1;
	for(i=1;i<n;i++){
		read(k); int x;
		while (k--)
			read(x),insert(x,n+i,1);
	}
	S=2*n,T=2*n+1;
	for(i=1;i<=n;i++) insert(S,i,1);
	for(i=1;i<n;i++) insert(n+i,T,1);
	maxflow();
	if (cnt<n) printf("-1"); else {
		for(i=1;i<n;i++) printf("%d %d\n",ans[i][0],ans[i][1]);
	}
}