洛谷 P1901 发射站

原题：
题目描述
某地有 N 个能量发射站排成一行，每个发射站 i 都有不相同的高度 Hi，并能向两边（两端的发射站只能向一边）同时发射能量值为 Vi的能量，发出的能量只被两边最近的且比它高的发射站接收。显然，每个发射站发来的能量有可能被 0 或 1 或 2 个其他发射站所接受。请计算出接收最多能量的发射站接收的能量是多少。

输入格式
第 1 行一个整数 N。

第 2 到 N+1 行，第 i+1 行有两个整数 Hi和 Vi，表示第 i 个发射站的高度和发射的能量值。

输出格式
输出仅一行，表示接收最多能量的发射站接收到的能量值。答案不超过 32 位带符号整数的表示范围。

输入
3
4 2
3 5
6 10

输出
7

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 1;
int n;
int h[maxn], v[maxn];
int nex[maxn], pre[maxn];
int res[maxn];
int main() {
    ios::sync_with_stdio(false);
    while (cin >> n) {
        for (int i = 1; i <= n; i++) {
            cin >> h[i] >> v[i];
        }
        stack<pair<int, int>> sn, sp;
        memset(pre, 0, sizeof(pre));
        memset(nex, 0, sizeof(nex));
        memset(res, 0, sizeof(res));
        for (int i = n; i >= 1; i--) {
            while (!sn.empty() && sn.top().second < h[i]) {
                sn.pop();
            }
            nex[i] = sn.empty() ? 0 : sn.top().first;
            sn.push(make_pair(i, h[i]));
        }
        for (int i = 1; i <= n; i++) {
            while (!sp.empty() && sp.top().second < h[i]) {
                sp.pop();
            }
            pre[i] = sp.empty() ? 0 : sp.top().first;
            sp.push(make_pair(i, h[i]));
        }
        int ans = 0;
        for (int i = 1; i <= n; i++) {
            res[pre[i]] += v[i];
            res[nex[i]] += v[i];
        }
        for (int i = 1; i <= n; i++) {
            ans = max(ans, res[i]);
        }
        cout << ans << endl;
    }
    return 0;
}

思路：
利用单调栈，找到每个数向左和向右的第一个出现的最大值。
然后计算每个接收塔接受的总和即可。