[bzoj3932][CQOI2015]任务查询系统_主席树

任务查询系统 bzoj-3932 CQOI-2015

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

注释：$1\le n,m\le 10^5$。

想法：离散化之后lower_bound，主席树查询即可。

最后，附上丑陋的代码... ...

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,num;
int tot,len,x,y,z;
long long pre,a,b,c;
int s[5000010],l[5000010],r[5000010],root[100010],cnt[5000010];
long long sum[5000010];
vector<int>p[100010],q[100010];
void build(int &x,int y)
{
    x=++tot;
    l[x]=l[y];
    r[x]=r[y];
    sum[x]=sum[y];
    cnt[x]=cnt[y];
    s[x]=num;
}
void updata(int L,int R,int x,int &y,int v,int c)
{
    if(s[x]!=num)
    {
        build(y,x);
    }
    cnt[y]+=c;
    sum[y]+=1ll*c*v;
    if(L==R)
    {
        return ;
    }
    if(v<=mid)
    {
        updata(L,mid,l[x],l[y],v,c);
    }
    else
    {
        updata(mid+1,R,r[x],r[y],v,c);
    }
}
long long query(int L,int R,int x,int k)
{
    if(L==R)
    {
        return 1ll*L*k;
    }
    if(cnt[l[x]]>=k)
    {
        return query(L,mid,l[x],k);
    }
    else
    {
        return query(mid+1,R,r[x],k-cnt[l[x]])+sum[l[x]];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int mn=1e9;
    int mx=-1e9;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        mn=min(mn,x);
        mx=max(mx,y);
        p[x].push_back(z);
        q[y].push_back(z);
    }
    for(int i=mn;i<=mx;i++)
    {
        build(root[i],root[i-1]);
        num++;
        len=p[i].size();
        for(int j=0;j<len;j++)
        {
            updata(1,1e7,root[i],root[i],p[i][j],1);
        }
        len=q[i-1].size();
        for(int j=0;j<len;j++)
        {
            updata(1,1e7,root[i],root[i],q[i-1][j],-1);
        }
    }
    pre=1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%lld%lld%lld",&x,&a,&b,&c);
        int k=(int) 1+(a*pre+b)%c;
        if(cnt[root[x]]<=k)
        {
            pre=sum[root[x]];
            printf("%lld\n",pre);
        }
        else
        {
            pre=query(1,1e7,root[x],k);
            printf("%lld\n",pre);
        }
    }
}

小结：主席树好好玩... ...