[CQOI2015]任务查询系统 主席树

题目描述

感觉再写数据结构我要折寿了

以时间为版本，优先级为价值建立主席树
对于一个任务，我们只要在任务开始的时候把它加进去，再在任务结束之后把它删除，其他时候只要继承上一个时间段的就好了，所以应该是很好实现的，差分然后求前缀和就好

跑得很慢内存很大的代码（懒得优化了）

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#define ll long long
using namespace std;
const int N=100010;
ll m,n,Pre=1,cnt,a[N];

struct array{ll e,v,flag;} b[N<<2];
void build(ll e,ll v,ll flag){b[++cnt].e=e;b[cnt].v=v;b[cnt].flag=flag;}
bool cmp(array x,array y) {return x.e < y.e;}
struct node
{
    node* ch[2];ll s,sum;
    void maintain()
    {
        s=0,sum=0;
        if(ch[0] != NULL) s+=ch[0]->s,sum+=ch[0]->sum;
        if(ch[1] != NULL) s+=ch[1]->s,sum+=ch[1]->sum;
    }
};
node* T[N];

void updata(node *pre,node* &o,int l,int r,int x,ll v,ll flag) 
//flag=1时加入任务，-1时删除任务
{
    if(l > r) return ;o=new node();
    int mid=(l+r)>>1;
    if(pre == NULL)
    {
        if(l == r) {o->s=flag;o->sum=v*flag;return ;}
        if(x > mid) updata(pre,o->ch[1],mid+1,r,x,v,flag);
        else updata(pre,o->ch[0],l,mid,x,v,flag);
        o->maintain();return ;
    }
    o->ch[0]=pre->ch[0],o->ch[1]=pre->ch[1];
    o->s=pre->s,o->sum=pre->sum;
    if(l == r){o->s+=flag;o->sum+=v*flag;return ;}
    if(x > mid) updata(pre->ch[1],o->ch[1],mid+1,r,x,v,flag);
    else updata(pre->ch[0],o->ch[0],l,mid,x,v,flag);
    o->maintain();
}

ll query(node* o,int l,int r,int k)
{
    if(l > r || o == NULL) return 0;
    if(o->s <= k) return o->sum;
    if(l == r) return a[l]*k;
    int mid=(l+r)>>1,ss=(o->ch[0] == NULL ? 0 : o->ch[0]->s);
    if(ss >= k) return query(o->ch[0],l,mid,k);
    return query(o->ch[0],l,mid,ss)+query(o->ch[1],mid+1,r,k-ss);
}

ll read()
{
    ll out =0,f=1;char c=getchar();
    while(c < '0' || c > '9') {if(c == '-') f=-1; c=getchar();}
    while(c >= '0' && c <= '9') {out=(out<<1)+(out<<3)+c-'0';c=getchar();}
    return out*f;
}

void solve()
{
    m=read(),n=read();
    for(int i=1;i<=m;i++)
    {
        ll s=read(),e=read(),v=read();
        build(s,v,1);build(e+1,v,-1);a[i]=v;
    }
    sort(a+1,a+m+1);sort(b+1,b+cnt+1,cmp);
    int j=1;
    for(int i=1;i<=n;i++)
    {
        T[i]=T[i-1];
        while(j <= cnt && b[j].e == i)
        {
            int x=lower_bound(a+1,a+1+m,b[j].v)-a;
            updata(T[i],T[i],1,m,x,b[j].v,b[j].flag);
            j++;
        }
    }
    for(int i=1;i<=n;i++)
    {
        ll x=read(),ai=read(),bi=read(),ci=read(),k=(ai*Pre+bi)%ci+1;
        printf("%lld\n",Pre=query(T[x],1,m,k));
    }
}

int main()
{
    solve();
    return 0;
}