洛谷P1198（线段树）

题目链接： https://www.luogu.org/problemnew/show/P1198

题目描述

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

1、 查询操作。

语法：Q L

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

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

2、 插入操作。

语法：A n

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

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

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

输入输出格式

输入格式：

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

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

输出格式：

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

思路：这道题初始没有给数，不放建一棵值均为零的树，每次在树后更新（插入）节点，当然最重要的是上板子（dalao们请跳过~），还要注意L=0的情况（我被这个坑了半小时······）

上代码

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
struct node
{
    int left, right;
    ll maxx, sum;
};

node tree[1000000];
//ll a[maxn];
int n;
ll k;
int M;
ll D;
int p, q;
char str[10];

void build(int m, int l, int r)
{
    tree[m].left = l;
    tree[m].right = r;
    if (l == r){
        tree[m].maxx = 0;
        tree[m].sum = 0;
        return;
    }
    int mid = (l + r) >> 1;
    build(m << 1, l, mid);
    build(m << 1 | 1, mid + 1, r);
    tree[m].maxx = max(tree[m << 1].maxx, tree[m << 1 | 1].maxx);
    tree[m].sum = tree[m << 1].sum + tree[m << 1 | 1].sum;
}

void update(int m, int a, ll val)
{
    if (tree[m].left == a && tree[m].right == a){
        tree[m].maxx += val;
        tree[m].sum += val;
        return;
    }
    int mid = (tree[m].left + tree[m].right) >> 1;
    if (a <= mid){
        update(m << 1, a, val);
    }
    else{
        update(m << 1 | 1, a, val);
    }
    tree[m].maxx = max(tree[m << 1].maxx, tree[m << 1 | 1].maxx);
    tree[m].sum = tree[m << 1].sum + tree[m << 1 | 1].sum;
}

ll querySum(int m, int l, int r)
{
    if (l == tree[m].left && r == tree[m].right){
        return tree[m].sum;
    }
    int mid = (tree[m].left + tree[m].right) >> 1;
    if (r <= mid){
        return querySum(m << 1, l, r);
    }
    else if (l > mid){
        return querySum(m << 1 | 1, l, r);
    }
    return querySum(m << 1, l, mid) + querySum(m << 1 | 1, mid + 1, r);
}

ll queryMax(int m, int l, int r)
{
    if (l == tree[m].left && r == tree[m].right){
        return tree[m].maxx;
    }
    int mid = (tree[m].left + tree[m].right) >> 1;
    if (r <= mid){
        return queryMax(m << 1, l, r);
    }
    else if (l > mid){
        return queryMax(m << 1 | 1, l, r);
    }
    return max(queryMax(m << 1, l, mid), queryMax(m << 1 | 1, mid + 1, r));
}

int main()
{
	int u=0;
	ll t=0;
	build(1,1,200010);
	scanf("%d%lld",&M,&D);
	for(int i=1;i<=M;i++)
	{
	  scanf("%s %lld",str,&k);
	  if(str[0]=='A')
	  {
	  	update(1,u+1,(k+t)%D);
	  	u++;
	  }
	  if(str[0]=='Q')
	  {
		  if(k==0)
		  {
		     t=0;
			 printf("0\n");	
		  }
		  else
		  {
		  t=queryMax(1,u-k+1,u);
	  	  printf("%lld\n",t);
	      }
	  }
    }
    return 0;
}