[CQOI2015] 任务查询系统
题目描述
最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。
超级计算机中的任务用三元组 ( s i , e i , p i ) (s_i, e_i, p_i) (si,ei,pi) 描述, ( s i , e i , p i ) (s_i, e_i, p_i) (si,ei,pi) 表示任务从第 s i s_i si 秒开始,在第 e i e_i ei 秒后结束(第 s i s_i si 秒和 e i e_i ei 秒任务也在运行),其优先级为 p i p_i pi。同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同。
调度系统会经常向查询系统询问,第 x i x_i xi 秒正在运行的任务中,优先级最小的 k i k_i ki 个任务(即将任务按照优先级从小到大排序后取前 k i k_i ki 个)的优先级之和是多少。
特别的,如果 k i k_i ki 大于第 x i x_i xi 秒正在运行的任务总数,则直接回答第 x i x_i xi 秒正在运行的任务优先级之和。上述所有参数均为整数,时间的范围在 [ 1 , n ] [1, n] [1,n] 之间。
输入格式
输入文件第一行包含两个空格分开的正整数 m m m 和 n n n,分别表示任务总数和时间范围。
接下来 m m m 行,每行包含三个空格分开的正整数 s i , e i , p i s_i,e_i,p_i si,ei,pi( s i ≤ e i s_i \le e_i si≤ei),描述一个任务。
接下来 n n n 行,每行包含四个空格分开的整数 x i , a i , b i , c i x_i,a_i,b_i,c_i xi,ai,bi,ci,描述一次查询。
本题强制在线。查询的参数 k i k_i ki 需要由公式 k i = 1 + ( a i × pre + b i ) m o d c i k_i = 1 +(a_i \times \text{pre}+b_i) \bmod c_i ki=1+(ai×pre+bi)modci 计算得到。其中 pre \text{pre} pre 表示上一次查询的结果,定义初始 pre = 1 \text{pre} = 1 pre=1 。
输出格式
输出共 n n n 行,每行一个整数,表示查询结果。
样例 #1
样例输入 #1
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
样例输出 #1
2
8
11
提示
【样例解释】
k 1 = ( 1 × 1 + 3 ) m o d 2 + 1 = 1 k_1 = (1\times 1 + 3)\bmod 2 + 1 = 1 k1=(1×1+3)mod2+1=1;
k 2 = ( 1 × 2 + 3 ) m o d 4 + 1 = 2 k_2 = (1\times 2+3)\bmod 4 + 1 = 2 k2=(1×2+3)mod4+1=2;
k 3 = ( 2 × 8 + 4 ) m o d 3 + 1 = 3 k_3 = (2 \times 8+4)\bmod 3+1 = 3 k3=(2×8+4)mod3+1=3。
【数据范围】
对于 100 % 100\% 100% 的数据, 1 ≤ m , n , c i , a i , b i ≤ 1 0 5 1\le m,n,c_i,a_i,b_i \le 10 ^ 5 1≤m,n,ci,ai,bi≤105, 1 ≤ s i ≤ e i ≤ n 1\leq s_i\leq e_i\leq n 1≤si≤ei≤n, 1 ≤ p i ≤ 1 0 7 1\le p_i \le 10^7 1≤pi≤107, x i x_i xi 为 1 1 1 到 n n n 的一个排列。
解析:
( s , e , p ) (s,e,p) (s,e,p) 可以看成是区间修改
查询 ( x , k ) (x,k) (x,k) 是单点查询,且查询最小的 k k k 个
主席树维护的是一个前缀和信息,所以可以考虑在主席树上差分,然后进行查询。
具体的,区间修改 ( s , e , p ) (s, e, p) (s,e,p) 可以通过差分转化成单点修改,即在 s s s 处的树上加上 p p p,在 e + 1 e+1 e+1 处的树上减去 p p p;对于查询,在主席树上进行查询即可
特别注意:
- 存在个数小于 k k k 的情况,需要特判一下
- 因为 p p p 的范围是 1e7,需要离散化
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
#define int ll
#define fi first
#define se second
#define debug(x) cerr << #x << ": " << (x) << endl
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
const int maxn = 1e5+10;
const int maxm = 1e5+10;
const int INF = 0x3f3f3f3f;
typedef pair<int, int> pii;
int m, n;
struct node{
int ls, rs;
int sum, siz;
}t[maxn << 6];
int tot;
vector<int> st[maxn], ed[maxn];
int a[maxn], root[maxn];
vector<int> b;
int ls(int rt){return t[rt].ls;}
int rs(int rt){return t[rt].rs;}
void modify(int &rt, int pre, int l, int r, int pos, int v){
rt = ++tot;
t[rt] = t[pre];
t[rt].siz += v;
t[rt].sum += v * b[pos-1];
if(l == r)
return;
int mid = (l+r) >> 1;
if(pos <= mid)
modify(t[rt].ls, ls(pre), l, mid, pos, v);
else
modify(t[rt].rs, rs(pre), mid+1, r, pos, v);
}
int query(int rt, int l, int r, int k){
if(l == r)
return t[rt].siz < k ? t[rt].sum : (t[rt].sum/t[rt].siz*k);
int mid = (l+r) >> 1;
int res = 0;
if(t[ls(rt)].siz >= k)
res = query(ls(rt), l, mid, k);
else
res = query(rs(rt), mid+1, r, k-t[ls(rt)].siz) + t[ls(rt)].sum;
return res;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> m >> n;
for(int i = 1; i <= m; i++){
int s, e;
cin >> s >> e;
st[s].push_back(i);
ed[e+1].push_back(i);
cin >> a[i];
b.push_back(a[i]);
}
sort(b.begin(), b.end());
b.erase(unique(b.begin(), b.end()), b.end());
int num = b.size();
for(int i = 1; i <= n; i++){
root[i] = root[i-1];
for(int j = 0; j < st[i].size(); j++){
int pos = lower_bound(b.begin(), b.end(), a[st[i][j]]) - b.begin() + 1;
modify(root[i], root[i], 1, num, pos, 1);
}
for(int j = 0; j < ed[i].size(); j++){
int pos = lower_bound(b.begin(), b.end(), a[ed[i][j]]) - b.begin() + 1;
modify(root[i], root[i], 1, num, pos, -1);
}
}
int pre = 1;
for(int i = 1; i <= n; i++){
int x, a, b, c, k;
cin >> x >> a >> b >> c;
k = (a * pre + b) % c + 1;
pre = query(root[x], 1, num, k);
cout << pre << endl;
}
return 0;
}