[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
;

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
;

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