HDU 4699 Editor(模拟 对顶栈)

题目大意:

给定一个整数序列 维护5种操作 次数<1e6

I x: 光标位置插入x 然后光标位于x之后

D: 删除光标前一个数

L: 光标左移

R: 光标右移

Q k: 询问位置k之前的最大前缀和

选用"对顶栈"的做法 大致示意图如下: 

 

对顶栈stkl, stkr直接通过数组模拟即可实现

以I x的操作为例, 需要:

1)将x插入stkl;

2)更新s数组 当前的前缀和 = 之前的前缀和 + x

3)更新f数组 记录该位置的最大前缀和

故有如下的代码:

// tl用于记录左栈的栈顶位置
void push_left(int x) {
    stkl[++tl] = x;
    s[tl] = s[tl-1] + x;
    f[tl] = max(f[tl-1], s[tl]);
}

对于D操作 直接将左栈栈顶出栈 即--tl;

对于L 左栈弹出 压入右栈

对于R 弹出右栈 剩下操作同I x的2) 3)

对于Q k return f[k]即可

完整代码如下:

#include <bits/stdc++.h>
using namespace std;

const int N = 1e8 + 10;

int stkl[N], stkr[N], tl, tr;
int s[N], f[N]; // s维护当前前缀和 f维护当前最大前缀和
char str[10];

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;
    while(~scanf("%d", &n)) {
        f[0] = INT_MIN;
        tl = tr = 0;
        s[0] = 0;
        while(n--) {
            scanf("%s", str);
            int x;
            if(*str == 'I') {
                scanf("%d", &x);
                push_left(x);
            }
            if(*str == 'D') {
                if(tl > 0) tl--;
            }
            if(*str == 'L') {
                if(tl > 0) stkr[++tr] = stkl[tl--];
            }
            if(*str == 'R') {
                if(tr > 0) push_left(stkr[tr--]);
            }
            if(*str == 'Q') {
                scanf("%d", &x);
                printf("%d\n", f[x]);
            }
        }
    }
}

这里注意一下每次的初始化即可

        f[0] = INT_MIN;
        tl = tr = 0;
        s[0] = 0;

猜你喜欢

转载自www.cnblogs.com/tedukuri/p/12374276.html
今日推荐