AcWing 编辑器
Description
你将要实现一个功能强大的整数序列编辑器。
在开始时,序列是空的。
编辑器共有五种指令,如下:
1、“I x”,在光标处插入数值x。
2、“D”,将光标前面的第一个元素删除,如果前面没有元素,则忽略此操作。
3、“L”,将光标向左移动,跳过一个元素,如果左边没有元素,则忽略此操作。
4、“R”,将光标向右移动,跳过一个元素,如果右边没有元素,则忽略次操作。
5、“Q k”,假设此刻光标之前的序列为a1,a2,…,an,输出,其中Si=a1+a2+…+ai
Input
第一行包含一个整数Q,表示指令的总数。
接下来Q行,每行一个指令,具体指令格式如题目描述。
Output
- 每一个“Q k”指令,输出一个整数作为结果,每个结果占一行。
Sample Input
8 I 2 I -1 I 1 Q 3 L D R Q 2
Sample Output
2 3
Data Size
- 1≤Q≤106,
|x|≤103,
1≤k≤n
题解:
- 栈。
- 还记得曾经动态求第k大时用的对顶堆吗?这题可以形象地理解为用“对顶栈”
- 设置两个栈,栈1存储光标左边的数,栈2存储光标右边的数。
- 光标左右移动就是将栈1/2的top弹掉,插入到另一个栈中。
- 删除就是直接弹掉栈1的top。
- 查询前缀和最大很容易。原理跟这篇这样。具体就是开两个数组sum和ans。sum实时维护栈1的前缀和,ans实时维护sum的最值。
#include <iostream>
#include <cstdio>
#include <stack>
#include <queue>
#define N 1000005
using namespace std;
int n;
stack<int> stk1, stk2;
int sum[N], ans[N] = {-0x7fffffff};
int read()
{
int x = 0, f = 1; char c = getchar();
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
return x *= f;
}
int main()
{
cin >> n;
for(int i = 1; i <= n; i++)
{
char s[3]; scanf("%s", s);
if(s[0] == 'I')
{
int x = read();
stk1.push(x);
int p = stk1.size();
sum[p] = sum[p - 1] + x;
ans[p] = max(ans[p - 1], sum[p]);
}
else if(s[0] == 'D' && stk1.size()) stk1.pop();
else if(s[0] == 'L' && stk1.size())
{
stk2.push(stk1.top());
stk1.pop();
}
else if(s[0] == 'R' && stk2.size())
{
stk1.push(stk2.top());
int p = stk1.size();
sum[p] = sum[p - 1] + stk2.top();
ans[p] = max(ans[p - 1], sum[p]);
stk2.pop();
}
else if(s[0] == 'Q')
{
int x = read();
printf("%d\n", ans[x]);
}
}
return 0;
}