题目链接
思路:模拟一下这个筛选的过程,每次用set的find找到最大的那个数,然后向左找k个。向右找k个,找到后把他们从Set里erase就行了,每次最大的那个数的话可以先排序后存队列里,每次取队首就是最大元素了。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
pair<int,int>v[maxn];
vector<int>p;
set<int>s;
queue<int>q;
bool cmp(const pair<int,int>&a,const pair<int,int>&b)
{
return a.first>b.first;
}
int main()
{
int n,k,cnt=1,ans[maxn];
scanf("%d %d",&n,&k);
for(int i=1;i<=n;++i)
scanf("%d",&v[i].first),v[i].second=i;
sort(v+1,v+1+n,cmp);
for(int i=1;i<=n;++i)
q.push(v[i].second),s.insert(i);
while(!s.empty())
{
p.clear();
while(!s.count(q.front())) q.pop();
int t=q.front();
q.pop();
auto it1=s.find(t),it2=s.find(t);
for(int i=0;i<=k;++i)
{
p.push_back(*it1);
if(it1==s.begin()) break;
--it1;
}
for(int i=0;i<k;++i)
{
++it2;
if(it2==s.end()) break;
p.push_back(*it2);
}
for(auto i:p)
{
s.erase(i);
ans[i]=cnt%2;
}
cnt++;
}
for(int i=1;i<=n;++i)
printf("%d",ans[i]==1?1:2);
}