[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);
        }
    }
}

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

posted @ 2018-07-24 22:11  JZYshuraK_彧  阅读(207)  评论(0编辑  收藏  举报