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("");
}