【題目敘述】http://codeforces.com/edu/course/2/lesson/7/2/practice/contest/289391/problem/B
【解題想法】並查集進階應用,尋找右邊(也可能是左邊)最近的目標
#include <iostream>
using namespace std;
int n, m, p[1000005];
int pa(int x){
if (x == p[x]) return x;
return p[x] = pa(p[x]);
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> n;
for (int i = 1; i <= n+1; i++){
p[i] = i;
}
for (int i = 0, x; i < n; i++){
cin >> x;
if (pa(x) == n+1){
cout << pa(1) << " ";
p[pa(1)] = pa(pa(1)+1);
}
else{
cout << pa(x) << " ";
p[pa(x)] = pa(pa(x)+1);
}
}
}