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 long
using 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;
}