【codeforces 1154E】stl使用

1.​​题目链接​​。题目大意：n个人分成两组，人的编号是1-n。每个人都有一个val。现在分组的规则是这样的，首先选区val最大的以及左右k个作为第一组，第二轮挑选也是从剩下的中间挑选出val最大的及其左右两个，现在问你最后的分组情况。

#include <bits/stdc++.h>
#include<queue>
using namespace std;
int main() {

  int n, k;
  cin >> n >> k;
  vector<pair<int, int>> a(n);
  for (int i = 0; i < n; ++i) {
    cin >> a[i].first;
    a[i].second = i;
  }
  sort(a.rbegin(), a.rend());
  queue<int> q;
  for (int i = 0; i < n; ++i) {
    q.push(a[i].second);
  }

  set<int> idx;
  for (int i = 0; i < n; ++i) {
    idx.insert(i);
  }
  string ans(n, '0');
  int who = 0;
  while (!idx.empty()) {
    while (!idx.count(q.front())) {
      q.pop();
    }
    int pos = q.front();
    q.pop();

    vector<int> add;
    auto it = idx.find(pos);
    for (int i = 0; i <= k; ++i) {
      add.push_back(*it);
      if (it == idx.begin()) break;
      --it;
    }
    it = next(idx.find(pos));
    for (int i = 0; i < k; ++i) {
      if (it == idx.end()) break;
      add.push_back(*it);
      ++it;
    }
    for (auto it : add) {
      idx.erase(it);
      ans[it] = '1' + who;
    }
    who ^= 1;
  }
  cout << ans << endl;

  return 0;
}