E. Two Teams

E. Two Teams

theme:n个人排成一行，每个人有一个属性值。先两个教练轮流选队员。每次从中选出属性值最大的人，并选出他左右连续的各k个人出来（如果有的话则出队），直至每个人都成为一对。求每个人最终属于哪个队。1<=n,k<=2e5

solution:模拟题。用双向队列模拟。首先预处理出每个位置紧挨着的前后下标，之后按属性从大到小排列，按l=L[l],r=R[r]左右遍历即可，每次遍历更新L[r]=l,R[l]=r

#include<bits/stdc++.h>
#include<vector>
#include<algorithm>
#define far(i,t,n) for(int i=t;i<n;++i)
#define pb(a) push_back(a)
#define lowbit(x) x&(-x)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int inf=0x3f3f3f3f;
int mod=998244353;
const int maxn=200010;

pair<int,int>a[maxn];
int L[maxn],R[maxn];

int ans[maxn];

int main()
{
    int n,k;
    cin>>n>>k;
    far(i,1,n+1)
    {
        scanf("%d",&a[i].first);
        a[i].second=i;
        L[i]=i-1;
        R[i]=i+1;
    }

    sort(a+1,a+n+1,greater<pair<int,int>>());
    int flag=1;
    far(i,1,n+1)
    {
        if(ans[a[i].second])
            continue;
        ans[a[i].second]=flag;
        int t=k,l,r=R[a[i].second];
        while(t--&&r<=n)
        {
            ans[r]=flag;
            r=R[r];
        }
        t=k,l=L[a[i].second];
        while(t--&&l>=1)
        {
            ans[l]=flag;
            l=L[l];
        }
        R[l]=r;
        L[r]=l;
        flag=flag==1?2:1;
    }
    far(i,1,n+1)
        printf("%d",ans[i]);
    puts("");
}