[CQOI2015]任务查询系统

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3352  Solved: 1072
[Submit][Status][Discuss]

Description

最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的

任务用三元组(Si,Ei,Pi)描述，(Si,Ei,Pi)表示任务从第Si秒开始，在第Ei秒后结束（第Si秒和Ei秒任务也在运行

），其优先级为Pi。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。调度系统会经常向

查询系统询问，第Xi秒正在运行的任务中，优先级最小的Ki个任务（即将任务按照优先级从小到大排序后取前Ki个

）的优先级之和是多少。特别的，如果Ki大于第Xi秒正在运行的任务总数，则直接回答第Xi秒正在运行的任务优先

级之和。上述所有参数均为整数，时间的范围在1到n之间（包含1和n）。

Input

输入文件第一行包含两个空格分开的正整数m和n，分别表示任务总数和时间范围。接下来m行，每行包含三个空格

分开的正整数Si、Ei和Pi(Si≤Ei)，描述一个任务。接下来n行，每行包含四个空格分开的整数Xi、Ai、Bi和Ci，

描述一次查询。查询的参数Ki需要由公式 Ki=1+(Ai*Pre+Bi) mod Ci计算得到。其中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

样例解释

K1 = (1*1+3)%2+1 = 1

K2 = (1*2+3)%4+1 = 2

K3 = (2*8+4)%3+1 = 3

对于100%的数据，1≤m,n,Si,Ei,Ci≤100000，0≤Ai,Bi≤100000，1≤Pi≤10000000，Xi为1到n的一个排列

思路

主席树+权值线段树

代码实现

#include<cstdio>
#define LL long long
const LL maxn=1e5+10;
inline LL min_(LL x,LL y){return x<y?x:y;}
inline LL max_(LL x,LL y){return x>y?x:y;}
LL n,m,ub,ans=1;
LL a,b,c,d;
LL es,e[maxn],ep[maxn<<1],en[maxn<<1];
LL h[maxn],hs,t[maxn<<7],s[maxn<<7],ls[maxn<<7],rs[maxn<<7];
char r_w[30],r_l;
LL in(LL&x){
    while(r_w[0]=getchar(),r_w[0]<'0'||r_w[0]>'9');r_l=1,x=0;
    while(r_w[r_l]=getchar(),r_w[r_l]>='0'&&r_w[r_l]<='9') r_l++;
    for(LL i=0;i<r_l;i++) x=x*10+r_w[i]-'0';
}
void out(LL x){
    if(!x) return;
    out(x/10);
    putchar(x%10+'0');
}
void add(LL pre,LL&suc,LL l,LL r,LL v){
    if(suc==pre) suc=++hs,t[suc]=t[pre],s[suc]=s[pre];
    if(v>0) t[suc]++;
    else t[suc]--;
    s[suc]+=v;
    if(l==r) return;
    if(!ls[suc]) ls[suc]=ls[pre];
    if(!rs[suc]) rs[suc]=rs[pre];
    LL mid=l+r>>1;
    if(max_(v,-v)<=mid) add(ls[pre],ls[suc],l,mid,v);
    else add(rs[pre],rs[suc],mid+1,r,v);
}
LL search(LL k,LL l,LL r,LL v){
    if(v<=0) return 0;
    if(v==t[k]) return s[k];
    if(l==r) return s[k]/t[k]*v;
    LL mid=l+r>>1,ret=0;
    ret+=search(ls[k],l,mid,min_(v,t[ls[k]]));
    ret+=search(rs[k],mid+1,r,min_(v-t[ls[k]],t[rs[k]]));
    return ret;
}
int main(){
    freopen("cqoi15_query.in","r",stdin);
    freopen("cqoi15_query.out","w",stdout);
    in(n),in(m);
    for(LL i=1;i<=n;i++){
        in(a),in(b),in(c),b++;
        ++es,ep[es]=c,en[es]=e[a],e[a]=es;
        ++es,ep[es]=-c,en[es]=e[b],e[b]=es;
        ub=max_(ub,c);
    }
    for(LL i=1;i<=m;i++){
        h[i]=h[i-1];
        for(LL k=e[i];k;k=en[k]) add(h[i-1],h[i],1,ub,ep[k]);
    }
    for(LL i=1;i<=m;i++){
        in(d),in(a),in(b),in(c);
        out(ans=search(h[d],1,ub,1+(ans*a+b)%c));
        if(!ans) putchar('0');
        putchar('
');
    }
    return 0;
}

感觉自己的代码常数好大呀。