任务查询系统 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);
}
}
}
小结：主席树好好玩... ...