最小树形图学习小计

问题来源

• JZOJ6997
• 裸的最小树形图

简单介绍

• 参考网上的博客。
• 假设是最小外向树。
• 简单记录：每一次将除了根节点以外的点找出最小的入边，这些边连成了一个环套树，对于所有环，将它们缩成一个点。对于所有点，将原本它的入边减去最小入边的权值，递归直到无环。
• 如果无根则多建立一个虚点作为根，并向每一个点连边，边权取决于是最小外向森林还是最小外向树。
• 边权的总和就是每一轮每一个点选择的最小入边的边权的和。
• 如果要还原这棵树，在最终图中有一条边插在环上，则将环上的这条边替换，对应了它的边权减去环边的这个操作。
• 正确性感受一下还是蛮显然的，时间复杂度是$O(nm)$

一个没有还原的代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 51
using namespace std;

int n,i,j,k,a[maxn],b[maxn],E[maxn][maxn],p[maxn],ans,P[maxn];
int E0[maxn][maxn],bz[maxn],fr[maxn],d[maxn],du[maxn];

int getans(int tot){
	memset(E,127,sizeof(E));
	for(i=1;i<=n;i++) for(j=1;j<=n;j++) if (i!=j)
		E[i][j]=b[i]^a[j];
	for(i=1;i<=n;i++) E[0][i]=a[i]^b[i];
	memset(bz,0,sizeof(bz));
	for(i=1;i<=tot;i++) bz[P[i]]=1;
	for(i=1;i<tot;i++) for(j=i+1;j<=tot;j++) E[P[j]][P[i]]=2e9;
	for(i=1;i<=tot;i++) for(j=1;j<=n;j++) if (!bz[j]) E[j][P[i]]=2e9;
	int ans=0,m=n;
	while (1){
		for(i=1;i<=m;i++){
			p[i]=0;
			for(j=1;j<=m;j++) if (E[j][i]<E[p[i]][i])
				p[i]=j;
			ans+=E[p[i]][i];
		}
		memset(bz,0,sizeof(bz)); int tp=0;
		memset(fr,0,sizeof(fr)); int cnt=0;
		memset(du,0,sizeof(du));
		for(i=1;i<=m;i++) du[p[i]]++;
		int t=0,w=0;
		for(i=1;i<=m;i++) if (du[i]==0) d[++w]=i;
		while (t<w){
			int x=d[++t]; bz[x]=1,fr[x]=++cnt;
			if (p[x]) if (!--du[p[x]]) d[++w]=p[x];
		}
		for(i=1;i<=m;i++) if (!bz[i]){
			tp=1,cnt++;
			int x=i; 
			while (!bz[x]) {
				bz[x]=1,fr[x]=cnt,x=p[x];
				int tmp=E[p[x]][x];
				for(int y=0;y<=m;y++) E[y][x]-=tmp;
			}
		} else {
			int tmp=E[p[i]][i];
			for(int y=0;y<=m;y++) E[y][i]-=tmp;
		}
		if (!tp) break;
		memset(E0,127,sizeof(E0));
		for(i=0;i<=m;i++) for(j=0;j<=m;j++) if (E[i][j]<2e9&&fr[i]!=fr[j])
			E0[fr[i]][fr[j]]=min(E0[fr[i]][fr[j]],E[i][j]);
		m=cnt,memcpy(E,E0,sizeof(E));
	}
	return ans;
}

int used[maxn];

int main(){
//	freopen("pl.in","r",stdin);
//	freopen("pl.out","w",stdout);
	scanf("%d",&n);
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
	for(i=1;i<=n;i++) scanf("%d",&b[i]);
	ans=getans(0);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) if (!used[j]){
			P[i]=j;
			if (getans(i)==ans){
				used[j]=1;break;
			}
		}
	}
	printf("%d\n",ans);
	for(int i=1;i<=n;i++) printf("%d ",P[i]);
}