D. Trash Problem

题目链接

题意

Vova决定打扫房间。可以将房间表示为坐标轴OX。房间里有n堆垃圾，第i堆的坐标是整数pi。所有桩具有不同的坐标。

让我们将总清理定义为以下过程。此过程的目标是在不超过两个不同的x坐标下收集所有桩。为了实现此目标，Vova可以执行几次（可能为零）移动。在移动期间，他可以选择一些x，并使用扫帚将所有堆从x移动到x + 1或x-1。请注意，他无法选择要移动多少个桩。

此外，有两种查询类型：

0 x-从坐标x移除一堆垃圾。确保此时在坐标x中有一个桩。
1 x-将一堆垃圾添加到坐标x。可以确保此时在坐标x中没有桩。
请注意，有时房间中的垃圾堆为零。

Vova如果想在进行任何查询之前进行彻底清理，则想知道他可以花费的最少移动次数。他还想在应用每个查询后知道此移动次数。查询以给定顺序应用。请注意，总清理实际上并没有发生，也不会改变堆的状态。它仅用于计算移动次数。

为了更好地理解，请阅读下面的“注释”部分以查看第一个示例的说明。

输入项
输入的第一行包含两个整数n和q（1≤n，q≤105）-分别是所有查询之前的房间中的桩数和查询数。

输入的第二行包含n个不同的整数p1，p2，…，pn（1≤pi≤109），其中pi是第i堆的坐标。

接下来的q行描述查询。第i个查询用两个整数ti和xi（0≤ti≤1;1≤xi≤109）描述，其中，如果需要从坐标xi中删除桩，则ti为0；如果需要添加，则为1。坐标xi的一堆。保证对于ti = 0，在当前桩组中有这样的桩，对于ti = 1，在当前桩组中没有这样的桩。

输出量
打印q + 1个整数：在第一个查询之前和q个查询中的每个查询之后，Vova进行总清理的最小移动次数。

思路1（线段树+离散化）218ms

$$我们可以观察到结果ans=p[n]-p[1]-max(p_j-p_i)(j=i+1,2<=j<=n)$$
$$所以接下来的问题就是维护数组的p[n],p[1],max(p_j-p_i)$$
$$因为n的数量会变大变小，我们可以预读取所有出现的权值组成一个数组$$
$$接着使用leap标记这个数在当前的数组中是否有效$$
$$题目数据范围是1-1e9，所以我们可以使用离散化+线段树维护$$
具体细节看代码的push_up函数就知道了

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
const int M = 1e5 + 5;
const int INF = 0x3f3f3f3f;
const int mod = 10007;
typedef long long ll;
struct point{
    int op,x;
}s[M];
int leap[N << 2],value[M];
int X[N << 2],tree[N << 2],mx[N << 2],mi[N << 2];
void push_up(int id)
{
	leap[id] = leap[id << 1] || leap[id << 1 | 1];
    if (leap[id << 1] && leap[id << 1 | 1]) mx[id] = max(mx[id << 1],mx[id << 1 | 1]);
    else if (leap[id << 1]) mx[id] = mx[id << 1];
    else if (leap[id << 1 | 1]) mx[id] = mx[id << 1 | 1];
 
    if (leap[id << 1] && leap[id << 1 | 1]) mi[id] = min(mi[id << 1],mi[id << 1 | 1]);
    else if (leap[id << 1]) mi[id] = mi[id << 1];
    else if (leap[id << 1 | 1]) mi[id] = mi[id << 1 | 1];
 
 
    if (leap[id << 1] && leap[id << 1 | 1]) tree[id] = max(max(tree[id << 1],tree[id << 1 | 1]),X[mi[id << 1 | 1]] - X[mx[id << 1]]);
    else if (leap[id << 1]) tree[id] = tree[id << 1];
    else if (leap[id << 1 | 1]) tree[id] = tree[id << 1 | 1];
    else tree[id] = 0;
  //  cout << "id = " << id << " leap=" << leap[id << 1] << " leap2=" << leap[id << 1 | 1] << " tree=" << tree[id];
   // cout << " mi=" << mi[id << 1 | 1] << " mx=" << mx[id << 1] << " X=" <<  X[mi[id << 1 | 1]] << " x2=" << X[mx[id << 1]] << endl;
}
void update(int index,int value,int l,int r,int id)
{
    if (l == r){
        mi[id] = index;
        mx[id] = index;
        leap[id] = value;
        return ;
    }
    int mid = (l + r) >> 1;
    if (index <= mid) update(index,value,l,mid,id << 1);
    if (index > mid) update(index,value,mid + 1,r,id << 1 | 1);
    push_up(id);
}
int main()
{
    int n,q;
    scanf("%d %d",&n,&q);
    for (int i = 0; i < n; i++){
        scanf("%d",&value[i]);
        X[i] = value[i];
    }
    int cnt = n;
    for (int i = 0; i < q; i++){
        scanf("%d %d",&s[i].op,&s[i].x);
        X[cnt++] = s[i].x;
    }
    sort(X,X + cnt);
    int len = unique(X,X + cnt) - X;
    for (int i = 0; i < n; i++){
        int index = lower_bound(X,X + len,value[i]) - X;
      //  cout << "index = " << index << endl;
        update(index,1,0,len - 1,1);
    }
    printf("%d\n",X[mx[1]] - X[mi[1]] - tree[1]);
    for (int i = 0; i < q; i++){
        int index = lower_bound(X,X + len,s[i].x) - X;
        update(index,s[i].op,0,len - 1,1);
        printf("%d\n",X[mx[1]] - X[mi[1]] - tree[1]);
    }
    return 0;
}

思路2（官方题解，使用set维护数组）1154ms

$$类似链表的插入删除维护数组和数组之间每两个相邻元素的差值$$

#include <bits/stdc++.h>
 
using namespace std;
 
int get(const set<int> &x, const multiset<int> &len) {
    if (len.empty()) return 0;
    return *x.rbegin() - *x.begin() - *len.rbegin();
}
 
void add(int p, set<int> &x, multiset<int> &len) {
    x.insert(p);
    auto it = x.find(p);
    int prv = -1, nxt = -1;
    if (it != x.begin()) {
        --it;
        len.insert(p - *it);
        prv = *it;
        ++it;
    }
    ++it;
    if (it != x.end()) {
        len.insert(*it - p);
        nxt = *it;
    }
    if (prv != -1 && nxt != -1) {
        len.erase(len.find(nxt - prv));
    }
}
 
void rem(int p, set<int> &x, multiset<int> &len) {
    auto it = x.find(p);
    int prv = -1, nxt = -1;
    if (it != x.begin()) {
        --it;
        len.erase(len.find(p - *it));
        prv = *it;
        ++it;
    }
    ++it;
    if (it != x.end()) {
        len.erase(len.find(*it - p));
        nxt = *it;
    }
    x.erase(p);
    if (prv != -1 && nxt != -1) {
        len.insert(nxt - prv);
    }
}
 
int main() {
#ifdef _DEBUG
    freopen("input.txt", "r", stdin);
//  freopen("output.txt", "w", stdout);
#endif
    
    int n, q;
    cin >> n >> q;
    set<int> x;
    multiset<int> len;
    for (int i = 0; i < n; ++i) {
        int p;
        cin >> p;
        add(p, x, len);
    }
    
    cout << get(x, len) << endl;
    for (int i = 0; i < q; ++i) {
        int t, p;
        cin >> t >> p;
        if (t == 0) {
            rem(p, x, len);
        } else {
            add(p, x, len);
        }
        cout << get(x, len) << endl;
    }
    
    return 0