BZOJ3932: [CQOI2015]任务查询系统

来源:互联网 发布:javascript效果源代码 编辑:程序博客网 时间:2024/05/02 05:06

传送门
题目大意:在[s,e](闭区间)上加入一个数字k(就相当于在s时刻放入,e+1时刻取出)。支持询问在t时刻时的第k大数。

想一想就可以知道主席树能水过这道题。

/**************************************************************    Problem: 3932    User: geng4512    Language: C++    Result: Accepted    Time:3808 ms    Memory:192900 kb****************************************************************/#include<cstdio>#include<vector>#include<algorithm>#define MAXN 100005#define LL long longusing namespace std;struct node {    int lc, rc, cnt;    long long sum;} t[MAXN*80];vector<int> L[MAXN], R[MAXN];int n, m, rt[MAXN], cnt, tag, mark[MAXN*80];char c;inline void GET(int &n) {    n = 0;    do c = getchar(); while('0' > c || c > '9');    do n = n*10+c-'0', c = getchar(); while('0' <= c && c <= '9');}inline void Build(int &rt, int x) {    t[rt = ++ cnt] = t[x];    mark[rt] = tag;}void Insert(int &rt, int p, int v, int l, int r, char f) {    if(mark[p] != tag) Build(rt, p);    t[rt].cnt += f;    t[rt].sum += v*f;    if(l >= r) return;    int mid = (l + r) >> 1;    if(v <= mid) Insert(t[rt].lc, t[p].lc, v, l, mid, f);    else Insert(t[rt].rc, t[p].rc, v, mid+1, r, f);}long long Query(int x, int l, int r, int k) {    if(l >= r) return 1LL * k * l;    int mid = (l + r) >> 1;    if(t[t[x].lc].cnt >= k) return Query(t[x].lc, l, mid, k);    return Query(t[x].rc, mid+1, r, k - t[t[x].lc].cnt) + t[t[x].lc].sum;}int main() {    int s, e, v, x, mn=MAXN, mx=-1;    long long p = 1;    scanf("%d%d", &n, &m);    for(int i = 1; i <= n; ++ i) {        GET(s); GET(e); GET(v);        L[s].push_back(v); R[e].push_back(v);        mn = min(mn, s); mx = max(mx, e);    }    for(int i = mn; i <= mx; ++ i) {        Build(rt[i], rt[i-1]); ++tag;        for(int j = 0; j < (int)L[i].size(); ++ j)            Insert(rt[i], rt[i], L[i][j], 1, 1e7, 1);        for(int j = 0; j < (int) R[i-1].size(); ++ j)            Insert(rt[i], rt[i], R[i-1][j], 1, 1e7, -1);    }    for(int i = 1; i <= m; ++ i) {        GET(x); GET(s); GET(e); GET(v);        v = 1 + (p*s+e)%v;        if(v >= t[rt[x]].cnt) p = t[rt[x]].sum;        else p = Query(rt[x], 1, 1e7, v);        printf("%lld\n", p);    }    return 0;}
1 0
原创粉丝点击