Codeforces Round #552 (Div. 3) E. Two Teams

题意：给出一个n和k，下面是n个数，要求找出最大的数，将它和周围k个数染色，第一次找到的染为1，第二次是2，第三次是1，类推。

思路：链表来解这个题思路很简单，不过不好写，好久没写过了。于是用并查集来解，感觉这种类型的题目并查集 都可以解。

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
typedef long long ll;
int n, a[N], pos[N], ans[N], l[N], r[N], t = 1, k;
int f1(int x) {
    return l[x] == x ? x : l[x] = f1(l[x]);
}
int f2(int x) {
    return r[x] == x ? x : r[x] = f2(r[x]);
}
int main() {
    cin >> n >> k;
    for(int i = 1; i <= n; i++)
        cin >> a[i], pos[a[i]] = i;
    for(int i = 1; i <= N; i++)
        l[i] = i, r[i] = i;
    for(int i = n; i > 0; i--) {
        int w = pos[i];
        if(ans[w])
            continue;
        ans[w] = t;
        int o = 0;
        for(int j = f1(w); j; j = f1(j)) {//cout<<j<<" ";
            if(ans[j] == 0) {
                ans[j] = t;
                o++;
            }
            l[j] = j - 1;
            if(o >= k)
                break;
        }
        o = 0;
        for(int j = f2(w); j <= n; j = f2(j)) {//cout<<j<<" ";
            if(ans[j] == 0) {
                ans[j] = t;
                o++;
            }
            r[j] = j + 1;
            if(o >= k)
                break;
        }
        t = (t == 1) ? 2 : 1;
    }
    for(int i = 1; i <= n; i++)
        cout << ans[i];
    return 0;
}