Codeforces 1180E Serge and Dining Room

題意
\(n\)個菜餚,有\(m\)個小朋友,每一個菜餚的價格爲\(a_i\),每一個小朋友有\(b_i\)元錢,小朋友從\(1 \rightarrow m\)依次購買菜餚,當第\(i\)個小朋友輪到的時候,他會購買他買的起的最貴的,不然就離開。
要求支持修改第\(i\)個菜餚的價格和修改第\(i\)個小朋友的擁有的錢數的兩種操做,每次操做完成給出\(m\)個小朋友買完後剩下的最貴的菜餚的價格是多少。node

思路
假設價格大於\(x\)\(y\)個菜餚都被買了,那麼顯然擁有錢數\(\geq x\)的小朋友個數必定要\(\geq y\),顯然如何購買是無所謂的。
那麼就在\(a_i\)處減一,\(b_i\)處加一,每次詢問一個最大的\(l\)使得\([l, \infty]\)的最大後綴和\(> 0\)
線段樹維護便可。c++

代碼spa

#include <bits/stdc++.h>
using namespace std;

#define N 1000010
#define M 1000000
int n, m, q, a[N], b[N];

struct SEG {
    struct node {
        int sum, Max;
        node() {
            sum = Max = 0;
        }
        node(int sum, int Max) : sum(sum), Max(Max) {}  
        node operator + (const node &other) const {
            node res = node();
            res.sum = sum + other.sum;
            res.Max = max(other.Max, Max + other.sum);
            return res;
        }
    }t[N << 2];
    void init() {
        memset(t, 0, sizeof t);
    }
    void update(int id, int l, int r, int x, int v) {
        if (l == r) {
            t[id].sum += v; 
            t[id].Max += v; 
            return;
        }   
        int mid = (l + r) >> 1;
        if (x <= mid) update(id << 1, l, mid, x, v);
        else update(id << 1 | 1, mid + 1, r, x, v);
        t[id] = t[id << 1] + t[id << 1 | 1];
    }
    int query(int id, int l, int r, node tmp) {  
        if (l == r) {
            return l;
        }
        int mid = (l + r) >> 1; 
        node tmp2 = t[id << 1 | 1] + tmp; 
        if (tmp2.Max > 0) {
            return query(id << 1 | 1, mid + 1, r, tmp);
        } else {
            return query(id << 1, l, mid, tmp2);
        }
    }
}seg;

int main() {
    while (scanf("%d%d", &n, &m) != EOF) {
        seg.init();
        for (int i = 1; i <= n; ++i) {
            scanf("%d", a + i);
            seg.update(1, 1, M, a[i], 1);
        }
        for (int i = 1; i <= m; ++i) {
            scanf("%d", b + i);
            seg.update(1, 1, M, b[i], -1);
        }
        scanf("%d", &q);
        int op, x, v;
        while (q--) {
            scanf("%d%d%d", &op, &x, &v);  
            switch(op) {
                case 1 :
                    seg.update(1, 1, M, a[x], -1);
                    seg.update(1, 1, M, a[x] = v, 1);
                    break; 
                case 2 :
                    seg.update(1, 1, M, b[x], 1);
                    seg.update(1, 1, M, b[x] = v, -1);
                    break;
                default :  
                    assert(0);
            }
            if (seg.t[1].Max <= 0) {
                puts("-1");
            } else {
                printf("%d\n", seg.query(1, 1, M, SEG::node(0, 0)));
            }
        }
    }
    return 0;
}
相關文章
相關標籤/搜索