Luogu 3586 [POI2015]LOG

考虑离散化后开权值线段树。

设序列中不小于$s$的数有$cnt$个，小于$s$的数的和为$sum$。

那么操作Z能成功的充要条件是$sum \geq (c - cnt) * s$。

如果序列中不小于$s$的数超过了$c$个，那么直接每一次都选这些数就好了。

如果没有超过$c$个的话，这$cnt$个数每一次都要选，然后再贪心地取剩下的数中最大的就行了。

需要把询问中出现的$a, s$和$0$一起离散化。

细节没想清楚WA好久

膜Claris

Code：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;

const int N = 1e6 + ;
const int inf =  << ;

int n, qn, tot = , maxn = , a[N];
ll b[N << ];

struct Querys {
    int type, x, y;
} q[N];

struct Innum {
    int val, id;
} in[N << ];

bool cmp(const Innum &x, const Innum &y) {
    if(x.val != y.val) return x.val < y.val;
    else return x.id < y.id;
}

inline int max(int x, int y) {
    return x > y ? x : y;
}

inline void read(int &X) {
    X = ;
    char ch = ;
    int op = ;
    for(; ch > '' || ch < ''; ch = getchar())
        if(ch == '-') op = -;
    for(; ch >= '' && ch <= ''; ch = getchar())
        X = (X << ) + (X << ) + ch - ;
    X *= op;
}

inline void discrete() {
    in[].val = -inf;
    sort(in + , in +  + tot, cmp);
    for(int cnt = , i = ; i <= tot; i++) {
        if(in[i].val != in[i - ].val) cnt++;
        maxn = max(maxn, cnt);
        b[cnt] = in[i].val;
        q[in[i].id].y = cnt;
    }
}

namespace SegT {
    ll s[N << ];
    int cnt[N << ];

    #define lc p << 1
    #define rc p << 1 | 1
    #define mid ((l + r) >> 1)

    inline void up(int p) {
        if(p) {
            s[p] = s[lc] + s[rc];
            cnt[p] = cnt[lc] + cnt[rc];
        }
    }

    void modify(int p, int l, int r, int x, int v) {
        if(x == l && r == x) {
            s[p] += (ll)b[l] * v;
            cnt[p] += v;
            return;
        }

        if(x <= mid) modify(lc, l, mid, x, v);
        else modify(rc, mid + , r, x, v);
        up(p);
    }

    ll qSum(int p, int l, int r, int x, int y) {
        if(x > y) return 0LL;
        if(x <= l && y >= r) return s[p];

        ll res = 0LL;
        if(x <= mid) res += qSum(lc, l, mid, x, y);
        if(y > mid) res += qSum(rc, mid + , r, x, y);
        return res;
    }

    int qCnt(int p, int l, int r, int x, int y) {
//        if(x > y) return 0;
        if(x <= l && y >= r) return cnt[p];

        int res = ;
        if(x <= mid) res += qCnt(lc, l, mid, x, y);
        if(y > mid) res += qCnt(rc, mid + , r, x, y);
        return res;
    } 

} using namespace SegT;

int main() {
    read(n), read(qn);
    in[++tot] = (Innum) {, };
    for(int i = ; i <= qn; i++) {
        char op[]; scanf("%s", op);
        read(q[i].x), read(q[i].y);
        if(op[] == 'U') q[i].type = , in[++tot] = (Innum) {q[i].y, i};
        else q[i].type = , in[++tot] = (Innum) {q[i].y, i};
    }

    discrete();

/*    for(int i = 1; i <= maxn; i++)
        printf("%lld ", b[i]);
    printf("\n");
    for(int i = 1; i <= qn; i++)
        printf("%d %d %d\n", q[i].type, q[i].x, q[i].y);
    printf("\n");   */

    for(int i = ; i <= n; i++)
        a[i] = , modify(, , maxn, , );

    for(int i = ; i <= qn; i++) {
        if(q[i].type == ) {
            modify(, , maxn, a[q[i].x], -);
            modify(, , maxn, q[i].y, );
            a[q[i].x] = q[i].y;
        } else {
            int cnt = qCnt(, , maxn, q[i].y, maxn);
/*            if(cnt >= q[i].x) {
                puts("TAK");
                continue;
            }   */
            ll sum = qSum(, , maxn, , q[i].y - );
            if(sum >= (q[i].x - cnt) * b[q[i].y]) puts("TAK");
            else puts("NIE");
        }
    }    

    return ;
}