P3168 [CQOI2015] 任务查询系统

[CQOI2015] 任务查询系统

题目描述

最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。

超级计算机中的任务用三元组 $(s_i,e_i,p_i)$ 描述，$(s_i,e_i,p_i)$ 表示任务从第 $s_i$ 秒开始，在第 $e_i$ 秒后结束（第 $s_i$ 秒和 $e_i$ 秒任务也在运行），其优先级为 $p_i$。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。

调度系统会经常向查询系统询问，第 $x_i$ 秒正在运行的任务中，优先级最小的 $k_i$ 个任务（即将任务按照优先级从小到大排序后取前 $k_i$ 个）的优先级之和是多少。

特别的，如果 $k_i$ 大于第 $x_i$ 秒正在运行的任务总数，则直接回答第 $x_i$ 秒正在运行的任务优先级之和。上述所有参数均为整数，时间的范围在 $[1,n]$ 之间。

输入格式

输入文件第一行包含两个空格分开的正整数 $m$ 和 $n$，分别表示任务总数和时间范围。

接下来 $m$ 行，每行包含三个空格分开的正整数 $s_i,e_i,p_i$（$s_i\le e_i$），描述一个任务。

接下来 $n$ 行，每行包含四个空格分开的整数 $x_i,a_i,b_i,c_i$，描述一次查询。

本题强制在线。查询的参数 $k_i$ 需要由公式 $k_i=1+(a_i\times \text{pre}+b_i)\mathrm{mod}{c}_i$ 计算得到。其中 $\text{pre}$ 表示上一次查询的结果，定义初始 $\text{pre}=1$ 。

输出格式

输出共 $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\times 1+3)\mathrm{mod}2+1=1$；

$k_2=(1\times 2+3)\mathrm{mod}4+1=2$；

$k_3=(2\times 8+4)\mathrm{mod}3+1=3$。

【数据范围】

对于 $100\%$ 的数据，$1\le m,n,c_i,a_i,b_i\le 10^5$，$1\le s_i\le e_i\le n$，$1\le p_i\le 10^7$，$x_i$ 为 $1$ 到 $n$ 的一个排列。

解析：

$(s,e,p)$ 可以看成是区间修改

查询 $(x,k)$ 是单点查询，且查询最小的 $k$ 个

主席树维护的是一个前缀和信息，所以可以考虑在主席树上差分，然后进行查询。

具体的，区间修改 $(s,e,p)$ 可以通过差分转化成单点修改，即在 $s$ 处的树上加上 $p$，在 $e+1$ 处的树上减去 $p$；对于查询，在主席树上进行查询即可

特别注意：

• 存在个数小于 $k$ 的情况，需要特判一下
• 因为 $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;
}