BZOJ 3932 任务查询系统

链接
主席树第二题。。。。
我是真的菜。
看一下问题，emmmmm
动态查询前K小
很明显就用主席树啦。
具体怎样实现呢？
以时刻为下标，优先级为区间建主席树。对于在一个区间[l,r]内存在的任务，在l处出现次数加1，在r+1处出现次数减1，把这些看作事件，将时刻i所有发生的时间加入i的线段树。然后就可以在T[x]上统计答案了。
给我们的时间不一定就是端点，所以要对输入的时间进行迁移。
查询的方法emmmmm
二分呗。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#define ls l,mid
#define rs mid+1,r
#define il inline
#define ll long long
using namespace std;
const ll maxm=300050;
ll rson[maxm<<5],lson[maxm<<5],size[maxm<<5],sum[maxm<<5];
ll root[maxm];
ll tot,max_w=-1;
ll to[maxm];
struct node{
    ll poi,val;
};
node e[maxm];
ll build(ll l,ll r)
{
    ll rt=++tot;
    size[rt]=sum[rt]=0;
    if(l<r)
    {
        ll mid=(l+r)>>1;
        lson[rt]=build(ls);
        rson[rt]=build(rs);
    }
    return rt;
}
ll updata(ll pre,ll l,ll r,ll val)
{
    ll rt=++tot;
    lson[rt]=lson[pre],rson[rt]=rson[pre];
    if(val>0) size[rt]=size[pre]+1;
    else size[rt]=size[pre]-1;
    sum[rt]=sum[pre]+val;
    if(l<r)
    {
       ll mid=(l+r)>>1;
       if(abs(val)<=mid) lson[rt]=updata(lson[pre],ls,val);
       else rson[rt]=updata(rson[pre],rs,val);
    }
    return rt;
}
bool comp(node x,node y)
{
    return x.poi<y.poi;
}
il ll get_ans(ll time,ll kth)
{
    ll x=root[time],l=1,r=max_w,ans=0;
    if(size[x]<=kth) return sum[x];
    while(l<r)
    {
        ll mid=(l+r)>>1;
        if(size[lson[x]]>=kth) r=mid,x=lson[x];
        else 
        {
            l=mid+1;
            kth-=size[lson[x]];
            ans+=sum[lson[x]];
            x=rson[x];
        }
    }
    if(kth) ans+=(1ll*l*kth);
    return ans;
}
int main()
{
    ll n,m;
    scanf("%lld%lld",&n,&m);
    ll cnt=0;
    for(ll i=1,t,s,p;i<=n;i++)
    {
        scanf("%lld%lld%lld",&t,&s,&p);
        e[++cnt].poi=t,e[cnt].val=p;
        e[++cnt].poi=s+1,e[cnt].val=-p;
        max_w=max(max_w,p);
    }
    //root[0]=build(1,max_w);
    sort(e+1,e+cnt+1,comp);
    for(ll i=1;i<=cnt;i++)
     root[i]=updata(root[i-1],1,max_w,e[i].val);
    for(ll i=cnt;i>=1;i--)  
     if(e[i].poi!=e[i+1].poi) to[e[i].poi]=i;  
    for(ll i=1;i<=m;i++)  
     if(!to[i]) to[i]=to[i-1];  
    ll pre=1;
    for(ll i=1;i<=m;i++)
    {
        ll x,a,b,c;
        scanf("%lld%lld%lld%lld",&x,&a,&b,&c);
        ll k=(a*pre+b)%c+1;
        pre=get_ans(to[x],k);
        printf("%lld\n",pre);
    }

    return 0;
}