洛谷P3168 任务查询系统

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

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

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

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

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

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

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

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

 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) >> ;
     if(p <= mid) {
         insert(ls[x], ls[y], p, v, l, mid);
     }
     else {
         insert(rs[x], rs[y], p, v, mid + , 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 ;
     }
     if(l == r) {
         return 1ll * std::min(k, sum[o]) * r;
     }
     int mid = (l + r) >> ;
     if(k <= sum[ls[o]]) {
         return ask(k, l, mid, ls[o]);
     }
     else {
         return Val[ls[o]] + ask(k - sum[ls[o]], mid + , r, rs[o]);
     }
 }

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

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

     LL lastans = ;
     for(int i = , 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 + ;
         lastans = ask(k, , lm, rt[t]);
         printf("%lld\n", lastans);
     }

     return ;
 }

AC代码