Educational Codeforces Round 23 D. Imbalanced Array 单调栈

链接：

http://codeforces.com/contest/817/problem/D

题意：

给你n个数a[1..n]定义连续子段imbalance值为最大值和最小值的差，要你求这个数组的imbalance总值

题解：

考虑每个位置作为最值向两边的最大拓展，最小值取负号，最大值取正号，求和即可。怎么去找呢？可以维护一个单调递增栈、单调递减栈

代码：

int n;
int a[MAXN];
int l_min[MAXN], r_min[MAXN], l_max[MAXN], r_max[MAXN];

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> n;
    rep(i, 1, n + 1) cin >> a[i];
    l_min[1] = l_max[1] = 0;
    rep(i, 2, n + 1) {
        l_min[i] = l_max[i] = i - 1;
        while (l_min[i] >= 1 && a[l_min[i]] > a[i])
            l_min[i] = l_min[l_min[i]];
        while (l_max[i] >= 1 && a[l_max[i]] < a[i])
            l_max[i] = l_max[l_max[i]];
    }
    r_min[n] = r_max[n] = n + 1;
    per(i, 0, n) {
        r_min[i] = r_max[i] = i + 1;
        while (r_min[i] <= n && a[r_min[i]] >= a[i])
            r_min[i] = r_min[r_min[i]];
        while (r_max[i] <= n && a[r_max[i]] <= a[i])
            r_max[i] = r_max[r_max[i]];
    }
    ll ans = 0;
    rep(i, 1, n + 1) {
        ans -= 1LL * a[i] * (i - l_min[i])*(r_min[i] - i);
        ans += 1LL * a[i] * (i - l_max[i])*(r_max[i] - i);
    }
    cout << ans << endl;
    return 0;
}