【bzoj3932】[CQOI2015]任务查询系统

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的一个排列

题解
主席树,要运用差分的思想把一个点分为(Si,Pi)与(Si+Ei,-Pi)
再根据时间点建立主席树就OK了,注意查询。

代码

#include
#include
#include
#include
#define ll long long
#define N 200010
#define M 10000010
ll sum[M];
int ls[M],rs[M],root[M],num[M],lim,to[M];
struct seg{int t;int v;}a[N];
int n,m,tot;
using namespace std;
ll read(){
    char ch=getchar();
    ll x=0,f=1;
    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;
}

bool cmp(seg a,seg b){
    return a.tint pre,int &k,int l,int r,int p){
    k=++tot;
    sum[k]=sum[pre]+p;
    if (p>0) num[k]=num[pre]+1;else num[k]=num[pre]-1;
    if (l==r) return;
    ls[k]=ls[pre];rs[k]=rs[pre];
    int mid=(l+r)/2;
    if (abs(p)<=mid) update(ls[pre],ls[k],l,mid,p);else update(rs[pre],rs[k],mid+1,r,p);
}

void init(){
    m=read();n=read();
    for (int i=1;i<=m;i++){
        int s=read(),e=read(),c=read();
        a[i].t=s;a[i].v=c;
        a[i+m].t=e+1;a[i+m].v=-c;
        lim=max(lim,c);
    }
    m*=2;
    sort(a+1,a+m+1,cmp);
    for (int i=1;i<=m;i++) {
        update(root[i-1],root[i],1,lim,a[i].v);
    }
    for (int i=m;i>=1;i--){
        if (a[i].t!=a[i+1].t) to[a[i].t]=i;
    }
    for (int i=1;i<=n;i++){
        if (to[i]==0) to[i]=to[i-1];
    }
}

ll query(int t,int k){
    int x=root[t],l=1,r=lim,mid;
    if (num[x]<=k) return sum[x];
    ll ans=0;
    while (l2;
        if (num[ls[x]]>=k){
            x=ls[x];
            r=mid;
        }else{
            k-=num[ls[x]];
            ans+=sum[ls[x]];
            x=rs[x];
            l=mid+1;
        }
    }
    if (k) ans+=(ll)(l)*(ll)(k);
    return ans; 
}

void solve(){
    ll pre=1;
    for (int i=1;i<=n;i++){
        ll x=read(),a=read(),b=read(),c=read();
        ll k=1+(a*pre+b)%c;
        pre=query(to[x],k);
        printf("%lld\n",pre);
    }
}

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

你可能感兴趣的:(主席树)