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;
}