任务查询系统(cqoi2015,bzoj3932)(主席树)

最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的
任务用三元组\((S_i,E_i,P_i)\)描述,\((S_i,E_i,P_i)\)表示任务从第\(S_i\)秒开始,在第\(E_i\)秒后结束(第\(S_i\)秒和\(E_i\)秒任务也在运行
),其优先级为\(P_i\)。同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同。调度系统会经常向
查询系统询问,第\(X_i\)秒正在运行的任务中,优先级最小的\(K_i\)个任务(即将任务按照优先级从小到大排序后取前\(K_i\)
)的优先级之和是多少。特别的,如果\(K_i\)大于第\(X_i\)秒正在运行的任务总数,则直接回答第\(X_i\)秒正在运行的任务优先
级之和。上述所有参数均为整数,时间的范围在1到\(n\)之间(包含1和\(n\))。

Input

输入文件第一行包含两个空格分开的正整数\(m\)\(n\),分别表示任务总数和时间范围。接下来\(m\)行,每行包含三个空格
分开的正整数\(S_i\)\(E_i\)\(P_i(S_i≤E_i)\),描述一个任务。接下来\(n\)行,每行包含四个空格分开的整数\(X_i\)\(A_i\)\(B_i\)\(C_i\)
描述一次查询。查询的参数\(K_i\)需要由公式 \(K_i=1+(A_i*Pre+B_i) % C_i\)计算得到。其中\(Pre\)表示上一次查询的结果,
对于第一次查询,\(Pre=1\)

Output

输出共n行,每行一个整数,表示查询结果。

Sample Input

4 3
1 2 6
2 3 3
1 3 2
3 3 4
3 1 3 2
1 1 3 4
2 2 4 3

Sample Output

2
8
11

HINT

样例解释

\(K_1 = (1*1+3)%2+1 = 1\)
\(K_2 = (1*2+3)%4+1 = 2\)
\(K_3 = (2*8+4)%3+1 = 3\)
对于100%的数据,\(1≤m,n,S_i,E_i,C_i≤100000\)\(0≤A_i,B_i≤100000\)\(1≤P_i≤10000000\),Xi为1到n的一个排列

题意:

中文题面,不解释

题解:

应该不难想到对于每一个点开一个权值线段树,但空间只有512MB,明显不行,而且如果我们把每个数暴力加入,肯定会TLE,所以需要一堆优化。
我们可以发现每个相隔的线段树,肯定有部分相同,所以我们可以用可持久化线段树来做了,同时对于一个数值,我们在左边那个时间点+1,在右边那个时间点往后一个点-1就可以了。我们开两个vector记录加减就行了。
好吧,还需要离散化。
最后我们对每一个询问求个区间和就行了。
这里有一个坑点,对于第K大那个点,我们需要加的不是这个值而是加上剩下的数乘上当前数。
然后就可以A掉了。

#include<bits/stdc++.h>
#define mid (l+(r-l)/2)
using namespace std;
const int N=100010;
int n,m,q;
vector<int>ad[N],dl[N];
int a[N];
int cnt,root[N],lson[N<<6],rson[N<<6],tr[N<<6];
long long sum[N<<6];
int build(int l,int r){
    int rt=++cnt;
    if(l==r)return rt;
    lson[rt]=build(l,mid);
    rson[rt]=build(mid+1,r);
    return rt;
}
inline void pushup(int rt){
    sum[rt]=sum[lson[rt]]+sum[rson[rt]];
}
int updata(int pre,int l,int r,int x,int z){
    int rt=++cnt;
    tr[rt]=tr[pre]+z;
    if(l==r){
        return rt;
    }
    lson[rt]=lson[pre];
    rson[rt]=rson[pre];
    if(x<=mid)lson[rt]=updata(lson[pre],l,mid,x,z);
    else rson[rt]=updata(rson[pre],mid+1,r,x,z);
    return rt;
}
long long query(int rt,int l,int r,int k){
    if(l==r)return a[l]*min(k,tr[rt]);
    if(tr[lson[rt]]>=k)return query(lson[rt],l,mid,k);
    else return query(rson[rt],mid+1,r,k-tr[lson[rt]])+sum[lson[rt]];
}
void dfs(int rt,int l,int r){
    if(!tr[rt]||sum[rt])return;
    if(l==r){
        sum[rt]=tr[rt]*a[l];
        return;
    }
    dfs(lson[rt],l,mid);
    dfs(rson[rt],mid+1,r);
    pushup(rt);
}
inline int read(){
    register int ret=0;register char c;
    for(c=getchar();!isdigit(c);c=getchar());
    for(;isdigit(c);ret=(ret<<1)+(ret<<3)+c-'0',c=getchar());
    return ret;
}
int main(){
    cin>>n>>q;
    for(int i=1;i<=n;++i){
        int l=read(),r=read(),p=read();
        ad[l].push_back(p);
        dl[r+1].push_back(p);
        a[i]=p;
    }
    sort(a+1,a+n+1);
    m=unique(a+1,a+n+1)-a-1;
    root[0]=build(1,m);
    for(register int i=1;i<=n;++i){
    	root[i]=root[i-1];
    	for(vector<int>::iterator it=ad[i].begin();it!=ad[i].end();++it){
    		root[i]=updata(root[i],1,m,lower_bound(a+1,a+m+1,*it)-a,1);
    	}
    	for(vector<int>::iterator it=dl[i].begin();it!=dl[i].end();++it){
    	    root[i]=updata(root[i],1,m,lower_bound(a+1,a+m+1,*it)-a,-1);
        }
    }
    for(int i=0;i<=n;++i){
        dfs(root[i],1,m);
    }
    long long last=1;
    while(q--){
        int x=read(),aa=read(),bb=read(),cc=read();
        aa=((long long)aa*(last%cc)+bb)%cc+1;
        last=query(root[x],1,m,aa);
        printf("%lld\n",last);
    }
}
posted @ 2018-12-11 09:53  整理者  阅读(202)  评论(0编辑  收藏  举报