洛谷P1198 [JSOI2008]最大数（线段树）

题目描述

现在请求你维护一个数列，要求提供以下两种操作：

1、 查询操作。

语法：Q L

功能：查询当前数列中末尾L个数中的最大的数，并输出这个数的值。

限制：LL不超过当前数列的长度。(L > 0)(L>0)

2、 插入操作。

语法：A n

功能：将nn加上tt，其中tt是最近一次查询操作的答案（如果还未执行过查询操作，则t=0t=0)，并将所得结果对一个固定的常数DD取模，将所得答案插入到数列的末尾。

限制：nn是整数（可能为负数）并且在长整范围内。

注意：初始时数列是空的，没有一个数。

输入输出格式

输入格式：

 

第一行两个整数，MM和DD，其中MM表示操作的个数(M \le 200,000)(M≤200,000)，DD如上文中所述，满足(0<D<2,000,000,000)(0<D<2,000,000,000)

接下来的MM行，每行一个字符串，描述一个具体的操作。语法如上文所述。

 

输出格式：

 

对于每一个查询操作，你应该按照顺序依次输出结果，每个结果占一行。

 

输入输出样例

输入样例#1： 

5 100
A 96
Q 1
A 97
Q 1
Q 2

输出样例#1： 

96
93
96

说明

[JSOI2008]

本题数据已加强

解题思路

建立区间为[1,m]的线段树，每个节点的权值表示区间[l,r]的最大值，非叶子节点的最大值就是max(左子树最大值，右子树最大值)。记录一下已经插入的数的数量，假如已经插入了cnt个数，那查询时就查询区间[cnt - L + 1, cnt]，插入时插入在l==r==cnt的位置。另外因为是取最大值，且存在负数，所以建树的时候要把各节点权值初始化为负无穷，但因为要模D，所以负无穷取值比-D小就可以了。

代码如下

#include <iostream>
#include <cstdio> 
#include <cstring>
#include <cmath>
#define INF -3000000000
using namespace std;
typedef long long ll;
struct T{
	ll l, r, maxx;
}tree[800005];
ll max(ll a, ll b)
{
	return a > b ? a : b;
}
ll query(ll k, ll l, ll r)
{
	if(tree[k].l >= l && tree[k].r <= r)
		return tree[k].maxx;
	ll mid = (tree[k].l + tree[k].r) / 2;
	ll left = INF;
	ll right = INF;
	if(l <= mid)
		left = query(2*k, l, r);
	if(r > mid)
		right = query(2*k+1, l, r);
	return max(left, right);
}
void update(ll k, ll p, ll z)
{
	if(tree[k].l == p && tree[k].r == p){
		tree[k].maxx = z;
		return;
	}
	ll mid = (tree[k].l + tree[k].r) / 2;
	if(p <= mid)
		update(2*k, p, z);
	if(p > mid)
		update(2*k+1, p, z);
	tree[k].maxx = max(tree[2*k].maxx, tree[2*k+1].maxx);
}
void build(int k, int l, int r)
{
	tree[k].l = l;
	tree[k].r = r;
	if(l == r){
		tree[k].maxx = INF;
		return;
	}
	build(2 * k, l, (l + r) / 2);
	build(2 * k + 1, (l + r) / 2 + 1, r);
	tree[k].maxx = max(tree[2 * k].maxx, tree[2*k+1].maxx);
}
int main()
{
	ll m, d;
	while(cin >> m >> d){
		build(1, 1, m);
		int t = 0;
		int cnt = 0;
		for(int i = 1; i <= m; i ++){
			string str;
			cin >> str;
			if(str == "A"){
				ll n;
				scanf("%lld", &n);
				n = (n + t) % d; 
				update(1, cnt + 1, n);
				cnt ++;
			}
			else {
				ll l;
				scanf("%lld", &l);
				t = query(1, cnt - l + 1, cnt);
				printf("%lld\n", t);
			}
		} 
	}
	return 0;
}