洛谷P3168 任务查询系统

题意：有n个任务，第i个的存在时间是l_i~r_i，有个权值。求t时刻第k大的权值。

这毒瘤...本来是前缀和 -> 主席树，我是树套树...然后光荣TLE。

其实很裸。一开始我写的是每个位置维护一个权值线段树。因为要片改点查，就用差分 + 树状数组搞定了。然后超时...

仔细思考，发现不带修可以直接用主席树。

把差分数组插入进去就行了，查询就是前缀和。这么一看好像主席树就是同时对多个数组做前缀和，每个数组就是线段树的一个位置。

我没用标记永久化，标记下传的时候直接新建的节点。

#include <cstdio>
#include <algorithm>
#include <vector>

typedef long long LL;
const int N = 100010, lm = 1e7, M = 20000010;

int n, m, tot, rt[N];
int sum[M], ls[M], rs[M];
LL Val[M];
std::vector<int> d[N];

void insert(int x, int &y, int p, int v, int l, int r) {
    if(!y || x == y) {
        y = ++tot;
        sum[y] = sum[x];
        Val[y] = Val[x];
    }
    if(l == r) {
        sum[y] += v;
        Val[y] += v * r;
        return;
    }
    if(!ls[y]) {
        ls[y] = ls[x];
    }
    if(!rs[y]) {
        rs[y] = rs[x];
    }
    int mid = (l + r) >> 1;
    if(p <= mid) {
        insert(ls[x], ls[y], p, v, l, mid);
    }
    else {
        insert(rs[x], rs[y], p, v, mid + 1, r);
    }
    sum[y] = sum[ls[y]] + sum[rs[y]];
    Val[y] = Val[ls[y]] + Val[rs[y]];
    return;
}

LL ask(int k, int l, int r, int o) {
    if(!o) {
        return 0;
    }
    if(l == r) {
        return 1ll * std::min(k, sum[o]) * r;
    }
    int mid = (l + r) >> 1;
    if(k <= sum[ls[o]]) {
        return ask(k, l, mid, ls[o]);
    }
    else {
        return Val[ls[o]] + ask(k - sum[ls[o]], mid + 1, r, rs[o]);
    }
}

int main() {
    scanf("%d%d", &n, &m);
    for(int i = 1, x, y, z; i <= n; i++) {
        scanf("%d%d%d", &x, &y, &z);
        d[x].push_back(z);
        d[y + 1].push_back(-z);
    }

    // prework
    for(int i = 1; i <= m; i++) {
        if(!d[i].size()) {
            rt[i] = rt[i - 1];
            continue;
        }
        for(int j = 0; j < d[i].size(); j++) {
            // d[i][j]
            if(d[i][j] > 0) {
                insert(rt[i - 1], rt[i], d[i][j], 1, 1, lm);
            }
            else {
                insert(rt[i - 1], rt[i], -d[i][j], -1, 1, lm);
            }
        }
    }

    LL lastans = 1;
    for(int i = 1, t, y, z, w; i <= m; i++) {
        scanf("%d%d%d%d", &t, &y, &z, &w);
        lastans %= w;
        y %= w;
        int k = (lastans * y + z) % w + 1;
        lastans = ask(k, 1, lm, rt[t]);
        printf("%lld
", lastans);
    }

    return 0;
}