E. Replace the Numbers

74 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

传送门

题意:

执行q次操作,有两个选项:添加一个数或者将当前所有指定的数变为另一个数。

思路:

我们记录每个数第一次出现的位置,作为“头”,在新添加数的时候,记录每个位置上的数第一次出现的位置,也就是它的“头”。在修改时,只需要合并“头”就行了。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
int ans[500010];
int link[500010];
int f[500100];//val
int find(int x)
{
//	link[x]==x?x:link[x]=find(link[x]);
	if(link[x] == x)return x;
	else
	{
		return link[x]=find(link[x]);
	}
}

int main()
{
	int q;
	cin>>q;
	int n = 0;
	while(q--)
	{
		int op;
		cin>>op;
		if(op == 1)
		{
			int x;
			cin>>x;
			ans[++n] = x;
			if(!f[x])f[x] = n,link[n] = n;
			else link[n] = f[x];
		}
		else
		{
			int x,y;
			cin>>x>>y;
			if(x==y || !f[x])continue;
			if(!f[y])
			{
				f[y] = f[x];
				ans[f[y]] = y;
				f[x] = 0;
			}
			else
			{
				if(f[x] < f[y])
				{
					ans[f[x]] = y;
					link[f[y]] = link[f[x]];
					f[y] = f[x];
					f[x] = 0;
				}
				else
				{
					link[f[x]] = f[y];
					ans[f[x]] = y;
					f[x] = f[y];
					f[x] = 0;
				}
			}
		}
	}
	for(int i = 1; i <= n; i++)
	{
		cout<<ans[f[ans[find(link[i])]]]<<" ";
	}
}