洛谷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

接下来的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 
#include  
#include 
#include 
#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;
} 

 

你可能感兴趣的:(洛谷,线段树树状数组)