Codeforces 1418 D. Trash Problem(set)

传送门

题意：

有 n n n堆垃圾，每次操作可以选择一个位置 x x x ,然后将垃圾移到 x + 1 x+1 x+1 或 x − 1 x-1 x−1 ,当剩下最多两堆垃圾时，表示清理完成。

有 m m m 次查询，查询分为两种：

1. 1. 1. 在位置 x x x 放一堆垃圾

2. 2. 2. 拿走位置 x x x 上的垃圾

每次查询要输出最少的操作次数去清理垃圾。

题解：

很明显，要将其分为两段，使得两段的长度之和最小。

一开始我是正着考虑维护答案，操作起来很麻烦。其实反着求更方便，即 答案 = = =两端点之差 − - − 中间最长的空隙。

那么可以想到用 s e t set set 维护。

代码：

#pragma GCC diagnostic error "-std=c++11"
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
const int mod = 1e9 + 7;
const int MAXN = 2e6 + 5;
const int inf = 0x3f3f3f3f;
set<int> s;
multiset<int> st;
void add(int x)
{
    s.insert(x);
    auto it = s.find(x);
    auto l = it, r = it;
    int flag = 0;
    if (l!=s.begin())
    {
        flag++;
        l--;
        st.insert(x - *l);
    }
    if(r!=--s.end())
    {
        flag++;
        r++;
        st.insert(*r - x);
    }
    if(flag==2)
        st.erase(st.find(*r - *l));
}   
void era(int x)
{
    auto it = s.find(x);
    auto l = it;
    auto r = it;
    int flag = 0;
    if(l!=s.begin()){
        flag++;
        l--;
        st.erase(st.find(x - *l));
    }
    if(r!=--s.end()){
        flag++;
        r++;
        st.erase(st.find(*r -x));
    }
    if(flag==2) {
        st.insert(*r - *l);
    }
    s.erase(it);
}
int query()
{
    if(s.size()<2)
        return 0;
    int c = *(--s.end()) - *s.begin();
    c -= *(--st.end());
    return c;
}
int main()
{
    // iss;
    // cin.tie(0);
    int n, q;
    cin >> n >> q;
    for (int i = 1; i <= n;i++){
        int x;
        cin >> x;
        add(x);
    }
    int ans = query();
    cout << ans << "\n";
    for (int i = 1; i <= q;i++){
        int op, x;
        cin >> op >> x;
        if(op==1){
            add(x);
        }
        else{
            era(x);
        }
        ans = query();
        cout << ans << endl;
    }
}

标签：include,int,auto,st,1418,++,flag,Problem,Trash
来源： https://blog.csdn.net/weixin_45755679/article/details/120571210