Codeforces Round #552 (Div. 3) E. Two Teams（set+模拟）

题目链接


思路：模拟一下这个筛选的过程，每次用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);
}