bzoj3932: [CQOI2015]任务查询系统

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3932

题解

　　我好笨啊这几天刷题好慢。
　　这道题是主席树题，每个任务就在开始的那一天的权值主席树上添加，在结束天的下一天的权值主席树上删掉。主席树维护size和sum，表示有多少数，以及这些数的和。
　　查询直接到对应树上找前K小的和就好了。

代码

//主席树#include <cstdio>#include <algorithm>#include <map>#define maxn 200010#define ll long longusing namespace std;ll N, M, tmp[maxn], s[maxn], e[maxn], p[maxn], ndtot, num[maxn], pos[maxn],    w[maxn], ans[maxn];struct segtree{    ll l, r, size;    ll sum;    segtree *ch[2];}pool[maxn*30], *root[maxn];inline ll read(ll x=0){    char c=getchar();    while(c<48 or c>57)c=getchar();    while(c>=48 and c<=57)x=(x<<1)+(x<<3)+c-48,c=getchar();    return x;}void ins(segtree *pre, segtree *now, ll v){    ll mid=(pre->l+pre->r)>>1;    *now=*pre; now->size+=v>0?1:-1,now->sum+=tmp[abs(v)]*(v>0?1:-1);    if(now->l==now->r)return;    if(abs(v)<=mid)ins(pre->ch[0],now->ch[0]=pool+ ++ndtot,v);    else ins(pre->ch[1],now->ch[1]=pool+ ++ndtot,v);}void build(segtree *p, ll l, ll r){    ll mid=(l+r)>>1;    p->l=l, p->r=r;    if(l==r)return;    build(p->ch[0]=pool+ ++ndtot,l,mid);    build(p->ch[1]=pool+ ++ndtot,mid+1,r);}ll sum(segtree *p, ll k){    ll ans=0;    if(k>=p->size)return p->sum;    if(p->l==p->r)return k*(p->sum/p->size);    ans+=sum(p->ch[0],k);    k-=p->ch[0]->size;    if(k>0)ans+=sum(p->ch[1],k);    return ans;}inline bool cmp(ll a, ll b){return pos[a]<pos[b];}void init(){    ll i, pt;    M=read(), N=read();    for(i=1;i<=M;i++)s[i]=read(), e[i]=read()+1, p[i]=read();    for(i=1;i<=M;i++)tmp[i]=p[i];    sort(tmp+1,tmp+M+1);    for(i=1;i<=M;i++)p[i]=lower_bound(tmp+1,tmp+M+1,p[i])-tmp;    for(i=1;i<=M;i++)    {        pos[i]=s[i], w[i]=p[i];        pos[i+M]=e[i], w[i+M]=-p[i];        num[i]=i, num[i+M]=i+M;    }    sort(num+1,num+M+M+1,cmp);    build(root[0]=pool+ ++ndtot,1,M+1);    for(i=1,pt=1;i<=N;i++)    {        for(root[i]=root[i-1];pos[num[pt]]==i;pt++)        {            segtree* p=root[i];            ins(p,root[i]=pool+ ++ndtot,w[num[pt]]);        }    }}void work(){    ll x, a, b, c, k;    ll lastans=1;    while(N--)    {        x=read(), a=read(), b=read(), c=read();        k=(a*lastans+b)%c+1;        printf("%lld\n",lastans=sum(root[x],k));    }}int main(){    init();    work();    return 0;}