题目描述
现在请求你维护一个数列,要求提供以下两种操作:
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; }