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

Posted 2021-03-12 arbor-one

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