Problem - I - Codeforces

看到最小值最大值，二分答案。

思路：每次二分时开两个集合，分别表示$0$颜色和$1$颜色。如果是$c$颜色，先将值存入$c$颜色，之后在$!c$颜色中找大于等于$mid-a$的值。

• 如果找到：删除$c$颜色中一个值为$a$的元素和$!c$颜色中一个值为大于等于$mid-a$的元素。
• 没有找到：找不大于$mid-a$中最大的值，判断该值是否存在，如果存在则返回$false$；否则继续处理。

对于C++的集合erase函数来说：

• set：set中的元素是不同的，非多重集。
• multiset：虽然元素可以是多个，但是按key删除是将集合中key相等的都删除。可以按迭代器删除。

这里提供另一种：将set中的key用元组表示，second用下标表示。这样即使first值相同，但是second值不同，实现了每次只删除一个元素。

代码如下：

#include "bits/stdc++.h"
using namespace std;
using LL = long long;
int main()
{int n; cin>>n;vector<int> a(n), c(n);for(auto &t: a) cin>>t;for(auto &t: c) cin>>t;LL l = 0, r = 1e9;auto check = [&](LL mid) -> bool {set<pair<LL,LL>> s[2];for(int i = 0; i < n; ++i) {s[c[i]].insert({a[i], i});if(s[!c[i]].size()) {auto pos = s[!c[i]].lower_bound({mid - a[i], -1});if(pos == s[!c[i]].end()) {auto it = pos--;if(it != s[!c[i]].begin()) {return false;}} else {s[!c[i]].erase(pos);s[c[i]].erase({a[i], i});}}}return true;};while(l < r) {LL mid = l + r + 1 >> 1;if(check(mid)) l = mid;else r = mid - 1;}cout<<l<<endl;
}