[CQOI2015]任务查询系统

Posted 2021-01-04 wolfycz

Description
最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的任务用三元组((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_ileqslant E_i))，描述一个任务。接下来n行，每行包含四个空格分开的整数(X_i)、(A_i)、(B_i)和(C_i)，描述一次查询。查询的参数(K_i)需要由公式 (K_i=1+(A_i imes Pre+B_i) mod 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 imes1+3)\%2+1 = 1)
(K_2 = (1 imes2+3)\%4+1 = 2)
(K_3 = (2 imes8+4)\%3+1 = 3)

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

主席树题，(P_i)需要离散化。建树的时候判断一下是根据root[i]还是root[i-1]进行更新，不插入点也要把之前的root信息copy一份。记录一下sum和cnt就ok了

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline int read(){
    int x=0,f=1;char ch=getchar();
    for (;ch<‘0‘||ch>‘9‘;ch=getchar())  if (ch==‘-‘)    f=-1;
    for (;ch>=‘0‘&&ch<=‘9‘;ch=getchar())    x=(x<<1)+(x<<3)+ch-‘0‘;
    return x*f;
}
inline void print(int x){
    if (x>=10)  print(x/10);
    putchar(x%10+‘0‘);
}
const int N=1e5,M=1e7;
int root[N+10],list[N+10];
struct S1{
    int x,v,type;
    void insert(int _x,int _v,int _type){x=_x,v=_v,type=_type;}
    bool operator <(const S1 &a)const{return x<a.x;}
}A[(N<<1)+10];
struct Segment{
    int ls[M+10],rs[M+10],cnt[M+10],tot;
    ll sum[M+10];
    void insert(int &k,int p,int l,int r,int x,int v){
        k=++tot;
        ls[k]=ls[p],rs[k]=rs[p];
        //记录cnt和sum
        cnt[k]=cnt[p]+v;
        sum[k]=sum[p]+1ll*list[x]*v;
        if (l==r)   return;
        int mid=(l+r)>>1;
        if (x<=mid) insert(ls[k],ls[p],l,mid,x,v);
        else    insert(rs[k],rs[p],mid+1,r,x,v);
    }
    ll Query(int k,int l,int r,int x){
        if (l==r)   return 1ll*list[l]*x;
        int mid=(l+r)>>1;
        if (x<=cnt[ls[k]])  return Query(ls[k],l,mid,x);
        else    return sum[ls[k]]+Query(rs[k],mid+1,r,x-cnt[ls[k]]);
    }
}Tree;
int main(){
    int n=read(),m=read();
    for (int i=1;i<=n;i++){
        int x=read(),y=read(),v=read();
        A[i].insert(x,v,1),A[i+n].insert(y+1,v,-1);
        list[i]=v;
    }
    sort(A+1,A+1+(n<<1));
    sort(list+1,list+1+n);
    int T=unique(list+1,list+1+n)-list-1;
    for (int i=1;i<=n<<1;i++)   A[i].v=lower_bound(list+1,list+1+T,A[i].v)-list;
    for (int i=1;i<=n<<1;i++){
        if (A[i-1].x!=A[i].x){//判断用谁更新
            for (int j=A[i-1].x+1;j<A[i].x;j++) root[j]=root[j-1];
            Tree.insert(root[A[i].x],root[A[i].x-1],1,T,A[i].v,A[i].type);
        }else   Tree.insert(root[A[i].x],root[A[i].x],1,T,A[i].v,A[i].type);
    }
    for (int j=A[n<<1].x+1;j<=m;j++)    root[j]=root[j-1];
    ll LastAns=1;
    for (int i=1;i<=m;i++){
        int x=read(),a=read(),b=read(),c=read(),k=(1ll*a*LastAns%c+b)%c+1;
        if (k>=Tree.cnt[root[x]])   printf("%lld
",LastAns=Tree.sum[root[x]]);
        else    printf("%lld
",LastAns=Tree.Query(root[x],1,T,k));
    }
    return 0;
}