Educational Codeforces Round 91 (Rated for Div. 2)E. Merging Towers 并查集

/*
 *1380E.Merging Towers
 *给出半径为1~n的n个盘子和m个塔，要求每个塔上盘子的半径始终从底向上递减
 *一次操作可以将一个塔上的任意个盘子移动到另一个塔的顶部。
 *令某一情形下的复杂度为将所有盘子移动到同一个塔上所需的最小操作数。
 *题目给出m-1次询问，每次询问时输出当前情形的复杂度并合并到两个塔的盘子到同一个塔上
 *考虑如果一个一个搬碟子，答案是n-1。
 *如果有两个相邻的碟子，就把他们看成一个整体，答案可以减去1.
 *按照每堆的堆顶建树，每次合并的时候，对于小树的节点我们都判断它在原数组的相邻节点的祖先节点是不是合并的大树top点
 *如果是，既然相邻那么他们的半径就是差值最小的，也就是说可以把他们看成一个整体 
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int fa[200005];
int p[200005];
vector<int>tree[200005];
int ff(int x){
    if(x==fa[x])return x;
    return fa[x]=ff(fa[x]);
} 
void solve(){
    int n,m;
    cin>>n>>m;
    ll ans=n-1;
    for(int i=1;i<=n;i++){
        fa[i]=i;
        int x;
        cin>>x;
        p[i]=x;
        if(p[i]==p[i-1])ans--;
        tree[x].push_back(i);
    }
    cout<<ans<<endl;
    for(int i=1;i<=m-1;i++){
        int x,y;
        cin>>x>>y;
        x=ff(x);
        y=ff(y);
        if(tree[x].size()>tree[y].size())swap(x,y);
        for(auto i:tree[x]){
            if(p[i+1]==y)ans--;
            if(p[i-1]==y)ans--;
        }
        for(auto i:tree[x]){
            p[i]=y;
            tree[y].push_back(i);
        }
        cout<<ans<<endl;
        fa[x]=y;
    }
}
int main(){
    int t=1;
    //cin>>t;
    while(t--){
        solve();
    }
}