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

题目链接：https://loj.ac/problem/516

题意：给出操作，将序列中所有一个数字替换为另一个，询问每次操作后距离最近的两个相同数字的距离。

解法：每个数字只与他的前驱和后继产生贡献。构建n个set，每次将较小的暴力合并到大的上面，通过lower_bound来找到他的前驱和后继。懒得离散化可以用map来存set。

#include <bits/stdc++.h>
using namespace std;
map <int, set<int>> mp;
int n, m, x, ans = 2147483647;
void update(int x, int y){
    set<int>::iterator it = mp[x].lower_bound(y);
    if(it != mp[x].end()) ans = min(ans, *it-y);
    if(it != mp[x].begin()) it--, ans = min(ans, y-*it);
}
int main()
{
    scanf("%d %d", &n,&m);
    for(int i=1; i<=n; i++){
        scanf("%d", &x);
        update(x, i);
        mp[x].insert(i);
    }
    for(int i=1; i<=m; i++){
        int a, b;
        scanf("%d %d", &a, &b);
        if(a == b){
            printf("%d\n", ans);
            continue;
        }
        if(mp[a].size()>mp[b].size()) swap(mp[a], mp[b]);
        for(set<int>::iterator it=mp[a].begin(); it!=mp[a].end(); it++){
            update(b, *it);
            mp[b].insert(*it);
        }
        mp[a].clear();
        printf("%d\n", ans);
    }
    return 0;
}