BZOJ 3932 - 主席树

传送门

题目分析

在只打会主席树模板的情况下做了这道题，也算是深有体会。
首先任务可以差分：一个任务是(s, e, p), 则在s处+1, 在e+1处-1，符合前缀。但是我们要查询指定时间的前k任务之和，可以想到主席树，且符合差分性质。
1~n每个节点代表从开始时间到现在，每个节点下是一颗权值线段树。用建立主席树的方法先将新时间节点置为前一个时间（从前一个时间更新），再将当前时间的所有差分操作更新（从当前时间更新），得到的每个时间节点便是从开始到此时间的一颗权值线段树。刚开始本来打的指针，调试被逼无奈改为数组（真·好写）。
建好树过后，对于查询（x, k), 就在x时间节点的这颗权值线段树上查询前k个元素之和，线段树的查询就不赘述了，详见代码。
另外，空间真的是个迷，卡了半天RE。

code

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

const int N = 1000005;
typedef long long ll;
int n, c[N * 4], lenc, m;
int p[N], taskCnt, maxx = -1, rt[N];
struct TASK{
    int point, tag, id;
    inline bool operator < (const TASK &b) const{
        return point < b.point;
    }
}task[N];
struct node{
    int lc, rc;
    ll sum, cnt;
    node(){}
}tr[N * 15];
int pool;
vector<int> lists[N];

inline int read(){
    int i = 0, f = 1; char ch = getchar();
    for(; (ch < '0' || ch > '9') && ch != '-'; ch = getchar());
    if(ch == '-') f = -1, ch = getchar();
    for(; ch >= '0' && ch <= '9'; ch = getchar()) i = (i << 3) + (i << 1) + (ch - '0');
    return i * f;
}

inline void wr(ll x){
    if(x < 0) putchar('-'), x = -x;
    if(x > 9) wr(x / 10);
    putchar(x % 10 + '0');
}

inline void disc_init(){
    sort(c + 1, c + lenc + 1);
    lenc = unique(c + 1, c + lenc + 1) - (c + 1);
    for(int i = 1; i <= m; i++) p[i] = lower_bound(c + 1, c + lenc + 1, p[i]) - c;
}

inline void insert(int x, int &y, int l, int r, int v, int cnt){
    tr[y = ++pool] = tr[x];
    tr[y].sum += cnt * c[v], tr[y].cnt += cnt;
    if(l == r) return;
    int mid = l + r >> 1;
    if(v <= mid) insert(tr[x].lc, tr[y].lc, l, mid, v, cnt);
    else insert(tr[x].rc, tr[y].rc, mid + 1, r, v, cnt);
}

inline ll query(int p, int l, int r, int k){
    if(tr[p].cnt <= k) return tr[p].sum;
    if(l == r) return c[l];
    int mid = l + r >> 1;
    if(k <= tr[tr[p].lc].cnt) return query(tr[p].lc, l, mid, k);
    else return query(tr[p].rc, mid + 1, r, k - tr[tr[p].lc].cnt) + tr[tr[p].lc].sum;
}

int main(){
    m = read(), n = read();
    for(int i = 1; i <= m; i++){
        task[++taskCnt].point = read(), task[taskCnt].tag = 1, task[taskCnt].id = i, lists[task[taskCnt].point].push_back(taskCnt);
        task[++taskCnt].point = read() + 1, task[taskCnt].tag = -1, task[taskCnt].id = i, lists[task[taskCnt].point].push_back(taskCnt);
        
        p[i] = c[++lenc] = read();
    }
    
    disc_init();
    
    for(int i = 1; i <= n; i++){
        insert(rt[i - 1], rt[i], 1, lenc, 0, 0);
        for(int j = 0; j < lists[i].size(); j++)
            insert(rt[i], rt[i], 1, lenc, p[task[lists[i][j]].id], task[lists[i][j]].tag);
    }

    ll pre = 1, ans;
    for(int i = 1; i <= n; i++){
        int x = read(), a = read(), b = read(), c = read();
        ans = query(rt[x], 1, lenc, ((a * pre + b) % c) + 1);
        wr(ans), putchar('\n');
        pre = ans;
    }
}

转载于:https://www.cnblogs.com/CzYoL/p/7643638.html