BZOJ 1345[BOI]序列问题 - 贪心 + 单调栈

题解

真的没有想到是单调栈啊。 回想起被单调栈支配的恐惧

最优情况一定是小的数去合并 尽量多的数，所以可以维护一个递减的单调栈。

如果加入的数比栈首小， 就直接推入栈。

如果加入的数大于等于栈首， 必须要合并栈首，因为栈首两边都是大的数， 要选择小的那个进行合并， 需要进行$st[top - 1] 与 x$判断大小。

最后留下来的递减数列逐个合并就可以了

代码

#include<cstring>
#include<cstdio>
#include<algorithm>
#define rd read()
using namespace std;

const int N = 1e6 + 1e5;

int st[N], tp, n;
long long ans;

int read() {
    int X = 0, p = 1; char c = getchar();
    for(; c > '9' || c < '0'; c = getchar()) if(c == '-') p = -1;
    for(; c >= '0' && c <= '9'; c = getchar()) X = X * 10 + c - '0';
    return X * p;
}

int main()
{
    n = rd;
    for(int i = 1; i <= n; ++i) {
        int x = rd;
        while(tp && st[tp] <= x) {
            if(tp > 1 && st[tp - 1] < x) ans += st[tp - 1];
            else ans += x;
            tp--;
        }
        st[++tp] = x;
    }
    while(tp > 1) {
        ans += st[--tp];
    }
    printf("%lld\n", ans);
}