题目链接:https://www.acwing.com/problem/content/130/
题意:你将要实现一个功能强大的整数序列编辑器。在开始时,序列是空的。编辑器共有五种指令,如下:
1、“I x”,在光标处插入数值x。
2、“D”,将光标前面的第一个元素删除,如果前面没有元素,则忽略此操作。
3、“L”,将光标向左移动,跳过一个元素,如果左边没有元素,则忽略此操作。
4、“R”,将光标向右移动,跳过一个元素,如果右边没有元素,则忽略次操作。
5、“Q k”,假设此刻光标之前的序列为a1,a2,…,an,输出max1≤i≤kSi,其中Si=a1+a2+…+ai。
数据范围
1≤Q≤106,
|x|≤103,
1≤k≤n
输入样例:
8
I 2
I -1
I 1
Q 3
L
D
R
Q 2
输出样例:
2
3
样例解释:
下图包含了对样例的过程描述:
思路:光标把序列分成了两段,我们可以用两个对顶的栈来实现这五个操作;当光标向左移动时,左边栈就相当于pop()一个顶部元素,右边栈就push()一个元素,光标右移类同;当查询前缀和的max时,相当于求左边栈前缀和的max,那么我们也可以再另开一个栈来存左边栈的最大前缀和(push()或pop()一个元素都不会影响前面的序列)。
代码实现:
#include<iostream>
#include<algorithm>
#include<limits.h>
using namespace std;
const int N = 1e6 + 5;
//s表示左侧栈的前缀和,f表示左侧栈前缀和max
int stkl[N], stkr[N], tl, tr;
int s[N], f[N];
void push_left(int x)
{
stkl[ ++ tl] = x;
s[tl] = s[tl - 1] + x;
f[tl] = max(f[tl - 1], s[tl]);
}
int main()
{
int n;
cin >> n;
char op;
f[0] = INT_MIN;
while (n -- ){
int x;
cin >> op;
if (op == 'I'){
cin >> x;
push_left(x);
}
else if (op == 'D'){
//栈顶--那么s也会--
if(tl > 0) tl -- ;
}
else if (op == 'L'){
if(tl > 0) stkr[ ++ tr] = stkl[tl -- ];
}
else if (op == 'R'){
if(tr > 0) push_left(stkr[tr -- ]);
}
else{
cin >> x;
cout << f[x] << endl;
}
}
return 0;
}