P1901 发射站

题目传送门

一、解题思路

1、又能向左发射，又能向右发射，不太好想。复杂问题都是由简单问题组装而来，我们先来考虑这个问题的一半。假如每个发射站只会向左发射信号，如果第$i$个发射塔比它左边的发射站都高，那么第$i$个发射站左边的站点就不可能收到$i$站点及$i$站后面所有站点发送来的信号，因为就算有，也被$i$拦截了。再保留这样的站点也就没有意义了，干掉就完了。

其实，本质上是在维护一个栈顶向栈底递增的单调栈，每次一个元素入栈之后，设此时栈顶为$tt$，那么该元素发射的信号只能被位于$tt-1$的站点收到，这样就起到了提速的作用。

2、向左发射信号的过程做完后再做一遍向右发射信号的过程。

3、找出哪个站点收到的信号最多。

二、数组模拟栈法

#include <bits/stdc++.h>

using namespace std;
const int N = 1000010;
int n;
int a[N], b[N]; //高度与能力值
int s[N];//结果数组
int ans; //最大值

//用数组模拟栈,每一个栈内的元素，都是需要找到“比自己高度小的,离自己最近”的发射塔，才能享受它带来的能量值。
int stk[N]; //内容：序号
int tt;     //指针,默认是0

void add(int i) {
    //把栈顶高度比自己小的出栈，这样的是不能享受i个发射塔带来的能量滴~
    while (tt && a[i] > a[stk[tt]]) tt--;
    //如果找着了一个高度比自己大的发射塔，那么，这个位置的发射塔将享受i个发射塔带来的能量。
    s[stk[tt]] += b[i];
    //把自己入栈，等待其它人给自己带来能量，enjoy!
    stk[++tt] = i;
}

int main() {
    //输入
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i] >> b[i];

    //从左向右扫一遍
    for (int i = 1; i <= n; i++) add(i);

    //还原指针，清空栈
    tt = 0;
    //从右向左扫一遍
    for (int i = n; i >= 1; i--) add(i);

    //找出结果
    for (int i = 1; i <= n; i++) ans = max(ans, s[i]);

    //输出
    printf("%d", ans);
    return 0;
}

三、STL栈法

#include <bits/stdc++.h>

using namespace std;
const int N = 1000010;

stack<int> stk; //栈
int a[N]; //高度数组
int b[N]; //能量数组
int s[N]; //每个发射塔的结果数组
int n;    //发射塔个数
int res;  //最大结果

/**
 * 功能：计算第i个发射塔，它可以给哪个发射塔带来能量
 * @param i
 */
void add(int i) {
    //把栈顶高度比自己小的出栈，这样的是不能享受i个发射塔带来的能量滴~
    while (!stk.empty() && a[i] > a[stk.top()]) stk.pop();
    //如果找着了一个高度比自己大的发射塔，那么，这个位置的发射塔将享受i个发射塔带来的能量。
    if (!stk.empty()) s[stk.top()] += b[i];
    //把自己入栈，等待其它人给自己带来能量，enjoy!
    stk.push(i);
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i] >> b[i];
    //从左向右扫一遍
    for (int i = 1; i <= n; i++) add(i);
    //清空栈
    while (!stk.empty()) stk.pop();
    //从右向左扫一遍
    for (int i = n; i >= 1; i--) add(i);
    //求最大值
    for (int i = 1; i <= n; i++) res = max(s[i], res);
    //输出
    printf("%d", res);
}