通过O(1)的时间实现栈的前缀和

通过O(1)的时间实现栈的前缀和

你将要实现一个功能强大的整数序列编辑器。

在开始时,序列是空的。

编辑器共有五种指令,如下:

1、“I x”,在光标处插入数值x。
2、“D”,将光标前面的第一个元素删除,如果前面没有元素,则忽略此操作。
3、“L”,将光标向左移动,跳过一个元素,如果左边没有元素,则忽略此操作。
4、“R”,将光标向右移动,跳过一个元素,如果右边没有元素,则忽略次操作。
5、“Q k”,假设此刻光标之前的序列为a1,a2,…,an,输出max1≤i≤kSi,其中Si=a1+a2+…+ai。

输入格式
第一行包含一个整数Q,表示指令的总数。

接下来Q行,每行一个指令,具体指令格式如题目描述。

输出格式
每一个“Q k”指令,输出一个整数作为结果,每个结果占一行。

数据范围
1≤Q≤106,
|x|≤103,
1≤k≤n
输入样例:
8
I 2
I -1
I 1
Q 3
L
D
R
Q 2
输出样例:
2
3
样例解释
下图包含了对样例的过程描述:

	#include<iostream>
	#include<algorithm>
	#include<stdio.h>
	#include<stack>
	using namespace std;
	char ch;
	const int N=1000100;
	stack<int>a,b;//建立两个栈,a用来存储上面输入的数,b用来储存光标右边的数
	int n;
	int sum[N],f[N];//sum储存的是这个下标下,的前缀和,f储存的是a的前缀和的最大值
	int main()
	{cin>>n;
	int x;
	f[0]=-1e7;
	 for(int i=1;i<=n;i++)
	 {  
	     cin>>ch;
	     if(ch=='I')
	     {
	         
	         cin>>x;
	         a.push(x);
	      sum[a.size()]=sum[a.size()-1]+a.top();
	     f[a.size()]=max(f[a.size()-1],sum[a.size()]);
	        
	     }
	     if(ch=='D')
	     {
	         if(!a.empty())
	         {
	            a.pop();
	         }
	     }
	     if(ch=='L')
	     {
	         if(!a.empty())
	       {
	           b.push(a.top()),a.pop();
	         
	       }
	     }   
	     if(ch=='R')
	     {   
	       if(!b.empty())
	      {  a.push(b.top()),b.pop();
	      sum[a.size()]=sum[a.size()-1]+a.top();
	     f[a.size()]=max(f[a.size()-1],sum[a.size()]);
	         }
	     }
	     if(ch=='Q')
	     { cin>>x;
	         cout<<f[x]<<endl;
	     }
	     getchar();
	 }
	  return 0;  
	}
posted @ 2020-02-02 12:03  arbor_one  阅读(181)  评论(0)    收藏  举报