洛谷P1198(线段树)

时间:2022-12-17 20:05:46

题目描述

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

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;
}