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

题目链接：https://codeforces.ml/contest/1154/problem/E

题意：有两个队伍1和2， 每次一个队伍选当前剩余的最大的数并且加上半径为k的范围内的数进自己的队伍，队伍1先手，求最后所有人的所属队伍

思路：考虑直接模拟，要nlogn的做法，而且题目给的是排列，那么考虑用set，直接用set操作 每个数的大小是实现不了的，因为set中是有序的，

想到set来存放 每个数的下标，要模拟的时候就半径为k的数都删掉即可，时间复杂度nlogn   因为不知道set红黑树是怎么实现的，所以不要边遍历的时候边删除

不然可能中途的it 发生变化导致错误，测试了一下样例不行    要拿vector把数都记录下来最后再一起删除

 

 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
const int mod=1e4;
const int maxn=2e5+10;
int n,k;
int p[maxn];
int mp[maxn];
int ans[maxn];


int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>n>>k;
    set<int>s;
    set<int>max1;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        p[i]=x;
        mp[x]=i;
        s.insert(i);
    }
    max1=s;
    vector<int>temp;
    int f=2;
    while(s.size())
    {
        temp.clear();
        int u=*max1.rbegin();
        int pos=mp[u];
        auto it=s.find(pos);
        temp.pb(pos);
        auto it1=it;
        auto it2=it;
        for(int i=1;i<=k;i++)
        {
            it2++;
            if(it2==s.end())
                break;
            temp.pb(*it2);
        }
        for(int i=1;i<=k;i++)
        {
            if(it1==s.begin())
                break;
            it1--;
            temp.pb(*it1);
        }
        for(auto &v:temp)
        {
            int num=1+f%2;
            ans[v]=num;
            int val=p[v];
            max1.erase(val);
            s.erase(v);
        }
        f++;
    }
    for(int i=1;i<=n;i++)
        cout<<ans[i];

    cout<<'\n';




}