Linking
题意:
两种操作:
- 1 x:在数列末尾加上数字x。 ( 1 ≤ x ≤ 5 ⋅ 1 0 5 ) (1≤x≤5⋅10^5) (1≤x≤5⋅105)
- 2 x y:将数列中的所有数字 x 全都换作 y。 ( 1 ≤ x , y ≤ 5 ⋅ 1 0 5 ) (1≤x,y≤5⋅10^5) (1≤x,y≤5⋅105)
现在给出 n n n ( 1 ≤ n ≤ 5 ⋅ 1 0 5 ) (1≤n≤5·10^5) (1≤n≤5⋅105) 次操作,输出最终的数列。
思路:
正着来的话,在更换数字这个操作就很麻烦,何况要一直更换。
所以考虑反着来,从后往前枚举所有操作:
在更换数字的这个操作中,将被更换的数字标记下来。这个加到数列中的数字就是更换过的。
需要注意的是,更换过之后还可能满足位置后面的更换条件,比如将2换作3之后要需要将3换作4,所以加到数列中的数字为更换成的最后一个。
我们定义
f
[
i
]
f[i]
f[i] 为数字i被更换成的最后一个数字,初始状态
f
[
i
]
=
i
f[i] = i
f[i]=i。
对于每次的更换数字操作:2 x y,把f[x]更新为f[y],即数字x更换成的最后一个数字为数字y更换成的最后一个数字。
每次添加操作直接在数组后面添加该数字被更换成的最后一个数字 f[x],最后将这个数组逆序输出。
Code:
const int N = 500010, mod = 1e9+7;
int T, n, m;
int a[N];
int k[N], p1[N], p2[N];
int f[N];
int main(){
Ios;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>k[i]>>p1[i];
if(k[i]==2) cin>>p2[i];
}
for(int i=1;i<=5e5+5;i++) f[i]=i;
int cnt=0;
for(int i=n;i>=1;i--)
{
if(k[i]==2)
{
int x=p1[i], y=p2[i];
f[x] = f[y];
}
else{
a[++cnt] = f[p1[i]];
}
}
for(int i=cnt;i>=1;i--) cout<<a[i]<<" ";
return 0;
}
经验:
思路很妙,正着不好处理,那就反过来看所有操作。
根据后面的操作来定下当前位置上是什么!
但是这个查找最后一个数字的操作不就是并查集的查询一个集合中的祖先节点的操作思路么?
但是为什么直接用并查集操作的话就会wa呢?搞不懂。
const int N = 500010, mod = 1e9+7;
int T, n, m;
int a[N];
int k[N], p1[N], p2[N];
int f[N];
int find(int x)
{
while(f[x]!=x) x=f[x];
return x;
}
int main(){
Ios;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>k[i]>>p1[i];
if(k[i]==2) cin>>p2[i];
}
for(int i=1;i<=5e5+5;i++) f[i]=i;
int cnt=0;
for(int i=n;i>=1;i--)
{
if(k[i]==2)
{
int x=p1[i], y=p2[i];
f[x] = find(y);
}
else{
a[++cnt] = find(p1[i]);
}
}
for(int i=cnt;i>=1;i--) cout<<a[i]<<" ";
return 0;
}