Codeforces Round #552 (Div. 3)-1154E-Two Teams-（模拟+双指针）

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<vector>
#include<iostream>
#include<cstring>
#include<set>
#include<queue>
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;

int n,k,maxxidx,maxx;
int a[200005];
int l[200005];///下标为i的人 的左边 下标是多少
int r[200005];///下标为i的人 的右边 下标是多少
int idx[200005];///标记原数组下标
int ans[200005];

void findmax()///找最大值，主要是改变全局变量maxxidx和maxx
{
    while(maxx>=1)
    {
        if( ans[ idx[maxx] ]==0 )
        {
            maxxidx=idx[ maxx ];
            break;
        }
        maxx--;
    }
}

int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        memset(a,0,sizeof(a));
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        memset(idx,0,sizeof(idx));
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            idx[ a[i] ]=i;
            l[i]=i-1;
            r[i]=i+1;
        }
        l[n+1]=n;
        r[0]=1;
        maxxidx=idx[n];///初始化为最大的下标
        maxx=n;
        int now=0;///当前已经选了多少人
        int t=1;///初始是1队先挑人
        while(now<n)
        {
            findmax();
            int i=maxxidx;///最大的那个人的下标
            ans[i]=t;
            now++;
            int x=l[i];
            int y=r[i];///中间向左右两边扩展
            for(int j=1;j<=k;j++)
            {
                if(x>=1 && x<=n)///如果x是0表明左边无人了
                    ans[x]=t,x=l[x],now++;///通过左右指针数组找下一个人
                if(y>=1 && y<=n)
                    ans[y]=t,y=r[y],now++;
            }
            ///挑完人就该改左右下标了
            l[y]=x;
            r[x]=y;
            if(t==1)
                t=2;
            else
                t=1;
        }
        for(int i=1;i<=n;i++)
            printf("%d",ans[i]);
        printf("\n");
    }
    return 0;
}