[启发式合并]#516. 「LibreOJ β Round #2」DP 一般看规律

[启发式合并]#516. 「LibreOJ β Round #2」DP 一般看规律

题目梗概

给定一个长度为 n 的序列 a，一共有 m 个操作。
每次操作的内容为：给定 x,y，序列中所有 x 会变成 y。

在每次操作后，求出相同元素的最近距离是多少。

解题思路

把一个数字变成另一个数字的过程看成两个坐标集合合并的过程。

用map离散之后，用set进行启发式合并就可以了。

在合并的同时更新答案。

#include<cstdio>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
inline int _read(){
    int num=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
const int maxn=300005;
map<int,int> id;
set<int> a[maxn];
int n,m,ans=2147483647,tot,cx[maxn],cy[maxn];
inline int h(int x){if (!id.count(x)) id[x]=++tot;return id[x];}
int min(int x,int y){if (x<y) return x;return y;}
void Insert(int x,int y){
    a[x].insert(y);
    set<int>::iterator t=a[x].find(y);
    set<int>::iterator t1=t;
    if (t!=a[x].begin()) t1--,ans=min(ans,*t-*t1);
    t1=t;t1++;
    if (t1!=a[x].end()) ans=min(ans,*t1-*t);
}
void merge(int &x,int &y){
    if (a[x].size()>a[y].size()) swap(x,y);
    while(!a[x].empty()){
        int w=*a[x].begin();
        Insert(y,w);
        a[x].erase(w);
    }
    printf("%d\n",ans);
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();m=_read();
    for (int i=1;i<=n;i++){int x=_read();Insert(h(x),i);}
    for (int i=1;i<=m;i++) cx[i]=_read(),cy[i]=_read(),h(cx[i]),h(cy[i]);
    for (int i=1;i<=m;i++) merge(id[cx[i]],id[cy[i]]);
    return 0;
}