BZOJ3932 [CQOI2015]任务查询系统 主席树+差分

给出M个任务,Q个查询,第 i 个任务从第 si 秒开始,到第 ti 秒结束,优先级为 pi .时间范围是N.要求支持查询 [sj,tj] 时间段内优先级最小的K个任务,查询强制在线。
M,N,Q<=100000.
注意到时间范围不大,因此对每个任务在时间上进行差分
主席树上每个节点存储当前时间的这一段的 size 和离散前的权值和 sum ,每次在第X颗主席树上查询前K个权值之和。

/**************************************************************
    Problem: 3932
    User: bdzxt
    Language: C++
    Result: Accepted
    Time:3132 ms
    Memory:60672 kb
****************************************************************/

#include
#define LL long long 
#define clr(x,i) memset(x,i,sizeof(x))
using namespace std;
const int N=100005;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct Node{
    int l,r,sz,sumv;
}t[N*36];
struct work{
    int t,v,f;
    friend bool operator <(work a,work b){
        return a.t*2];
int n,m,rk[N],sz,cnt,rt[N];
void upd(int l,int r,int &x,int y,int pos,int flag)
{   
    t[++cnt]=t[y];x=cnt;
    t[cnt].sz+=flag;t[cnt].sumv+=(LL)rk[pos]*flag;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(pos<=mid)
      upd(l,mid,t[x].l,t[y].l,pos,flag);
    else
      upd(mid+1,r,t[x].r,t[y].r,pos,flag);
}
LL query(int l,int r,int x,LL k)
{
    int nowsize=t[x].sz;
    if(k>=t[x].sz) return t[x].sumv;
    if(l==r) return rk[l]*k;
    int mid=(l+r)>>1;
    if(t[t[x].l].sz>=k)
      return query(l,mid,t[x].l,k);
    else
      return t[t[x].l].sumv+query(mid+1,r,t[x].r,k-t[t[x].l].sz);
}
int main()
{
    m=read();n=read();
    for(int i=1,s,t,v;i<=m;i++){
        s=read();t=read();v=read();
        a[i*2-1]=(work){s,v,1};
        a[i*2]=(work){t+1,v,-1};
        rk[i]=v;
    }
    sort(a+1,a+2*m+1);sort(rk+1,rk+m+1);
    sz=unique(rk+1,rk+m+1)-rk-1;
    m*=2;
    for(int i=1;i<=m;i++)
    {
        int x=lower_bound(rk+1,rk+sz+1,a[i].v)-rk,k=a[i].t,flag=a[i].f;
        if(k!=a[i-1].t)
        {
            if(i!=1) for(int j=a[i-1].t+1;j1];
            upd(1,sz,rt[k],rt[k-1],x,flag);
        }
        else upd(1,sz,rt[k],rt[k],x,flag);
    }
    LL pre=1;
    for(int i=1;i<=n;i++)
    {
        LL x=read(),a=read(),b=read(),c=read(),y;
        y=1+(a*pre+b)%c;
        pre=query(1,sz,rt[x],y);
        printf("%lld\n",pre);
    }
    return 0;
}

你可能感兴趣的:(差分,主席树,BZOJ)