栈的加强版
Description
请你实现一个加强版的栈,支持以下操作:
push x: 向栈顶加入一个整数x
pop: 从栈顶弹出一个整数,并且输出该整数
inc k x: 将处于栈底的前k个整数加x。
Input
第一行包含一个整数N,代表操作的数量。
以下N行每行一条操作。
1 ≤ N ≤ 200000, 0 ≤ x ≤ 100000, 1 ≤ k ≤ 当前栈的大小
Output
对于每一个pop操作,输出弹出的整数数值。
Sample Input
6
push 1
inc 1 2
push 2
inc 2 2
pop
pop
Sample Output
4
5
思路
我们对前k个整数加上x进行操作时,可能要TLE,因为如果inc这个操作多了起来,下面的k之前的每个数都需要加,那么我们需要找一个更优的方法,这时想到的是线段树的延迟更新,把这个思路移步过来即可,如果删除此节点那么将此位置的标记数加上, 并且传到下一个标记数,将则位置的标记数制为0。
代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<map>
#include<stack>
#include<queue>
#include<string>
#include<vector>
#include<utility>
#include<algorithm>
using namespace std;
const int MAXN=200000+5;
typedef long long ll;
typedef pair<int,int> pll;
ll add[MAXN];
int main(){
int n,cnt=1;
stack<pll> st;
cin>>n;
while(n--){
string s;
int x,k;
cin>>s;
if(s[1]=='u'){
cin>>x;
st.push(pll(x,cnt++));
}
if(s[1]=='n'){
cin>>x>>k;
add[x]+=k;
}
if(s[1]=='o'){
pll now=st.top();
st.pop();
int x=now.first;
int y=now.second;
cout<<x+add[y]<<endl;
add[y-1]+=add[y];
add[y]=0;
cnt--;
}
}
return 0;
}