题目描述
最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的任务用三元组(Si,Ei,Pi)描述，(Si,Ei,Pi)表示任务从第Si秒开始，在第Ei秒后结束（第Si秒和Ei秒任务也在运行），其优先级为Pi。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。调度系统会经常向查询系统询问，第Xi秒正在运行的任务中，优先级最小的Ki个任务（即将任务按照优先级从小到大排序后取前Ki个）的优先级之和是多少。特别的，如果Ki大于第Xi秒正在运行的任务总数，则直接回答第Xi秒正在运行的任务优先
级之和。上述所有参数均为整数，时间的范围在1到n之间（包含1和n）。
输入
输入文件第一行包含两个空格分开的正整数m和n，分别表示任务总数和时间范围。接下来m行，每行包含三个空格分开的正整数Si、Ei和Pi(Si≤Ei)，描述一个任务。接下来n行，每行包含四个空格分开的整数Xi、Ai、Bi和Ci，描述一次查询。查询的参数Ki需要由公式 Ki=1+(Ai*Pre+Bi) mod Ci计算得到。其中Pre表示上一次查询的结果，对于第一次查询，Pre=1。
输出
输出共n行，每行一个整数，表示查询结果。
样例输入
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
样例输出
2
8
11
题解
主席树（不需要修改）
先将所有任务读进来，将优先级离散化，再按时间排序加入主席树中。
开始时间加进来，结束时间+1减去。
查询时，相当于查询[1,x]这一段区间，所以不用变为两树相减的形式。
先二分查找出最后一个小于等于x的（即第一个大于x的数-1），然后查询。
注意：需要开long long；sum数组修改时要加减原值，不能是离散化后的值；查询时如果l和r相等，需要返回l的原值乘p
upperbound码错调了半天。。。
#include <cstdio>
#include <algorithm>
#define N 200010
using namespace std;
typedef long long lint;
struct data
{
lint ti , p , mk;
}a[N];
lint v[N] , top , root[N] , lp[N << 6] , rp[N << 6] , si[N << 6] , sum[N << 6] , tot;
lint vti[N];
bool cmp1(data a , data b)
{
return a.p < b.p;
}
bool cmp2(data a , data b)
{
return a.ti < b.ti;
}
void pushup(lint x)
{
si[x] = si[lp[x]] + si[rp[x]];
sum[x] = sum[lp[x]] + sum[rp[x]];
}
void update(lint x , lint &y , lint l , lint r , lint p , lint a)
{
y = ++tot;
if(l == r)
{
si[y] = si[x] + a;
sum[y] = sum[x] + v[p] * a;
return;
}
lint mid = (l + r) >> 1;
if(p <= mid) rp[y] = rp[x] , update(lp[x] , lp[y] , l , mid , p , a);
else lp[y] = lp[x] , update(rp[x] , rp[y] , mid + 1 , r , p , a);
pushup(y);
}
lint query(lint y , lint l , lint r , lint p)
{
if(l == r)
return v[l] * p;
lint mid = (l + r) >> 1;
if(p <= si[lp[y]]) return query(lp[y] , l , mid , p);
else return query(rp[y] , mid + 1 , r , p - si[lp[y]]) + sum[lp[y]];
}
int main()
{
lint n , m , i , x , y , z , pre = 1 , c , k , p;
scanf("%lld%lld" , &n , &m);
for(i = 1 ; i <= n ; i ++ )
{
scanf("%lld%lld%lld" , &x , &y , &z);
a[2 * i - 1].ti = x;
a[2 * i].ti = y + 1;
a[2 * i - 1].p = a[2 * i].p = z;
a[2 * i - 1].mk = 1;
a[2 * i].mk = -1;
}
sort(a + 1 , a + 2 * n + 1 , cmp1);
for(i = 1 ; i <= 2 * n ; i ++ )
{
if(a[i].p != v[top]) v[++top] = a[i].p;
a[i].p = top;
}
sort(a + 1 , a + 2 * n + 1 , cmp2);
for(i = 1 ; i <= 2 * n ; i ++ )
{
vti[i] = a[i].ti;
update(root[i - 1] , root[i] , 1 , top , a[i].p , a[i].mk);
}
while(m -- )
{
scanf("%lld%lld%lld%lld" , &c , &x , &y , &z);
k = 1 + (x * pre + y) % z;
p = upper_bound(vti + 1 , vti + 2 * n + 1 , c) - vti - 1;
if(si[root[p]] < k) pre = sum[root[p]];
else pre = query(root[p] , 1 , top , k);
printf("%lld
" , pre);
}
return 0;
}