G Tree Projection 2020ICPC·小米 网络选拔赛第一场

https://ac.nowcoder.com/acm/contest/7501/G

把b中的数字变成对应a中数字的哪一个位置,dy[a[i]]=i,b[i]=dy[b[i]]

然后从前往后枚举b[i],每次b[i]与之前出现最小的b[j]相连

因为b[i]是dfs序,且从左到右枚举b[i],那么这棵树从b[1]开始是满足dfs序的

又因为每次与之前出现最小的b[j]相连,且b[j]是原b[j]数字出现在a数组中的位置,越小越靠前,那么一定是b[i]=1也就是原b[i]=a[1]的地方为根节点,从上到下一定是满足拓扑序的

#include<bits/stdc++.h>
using namespace std;

const int maxl=2e5+10;

int n,m,tot;
int a[maxl],b[maxl],dy[maxl];
struct ed
{
	int u,v;
}c[maxl];

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&a[i]);
		dy[a[i]]=i;
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&b[i]);
		b[i]=dy[b[i]];
	}
}

inline void mainwork()
{
	int rt=b[1];
	for(int i=2;i<=n;i++)
	{
		c[++tot]=ed{rt,b[i]};
		if(b[i]<rt)
			rt=b[i];
	}
}

inline void print()
{
	puts("YES");
	for(int i=1;i<=tot;i++)
		printf("%d %d\n",a[c[i].u],a[c[i].v]);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/liufengwei1/article/details/109328587