题目链接: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;
}