Rockdu's Blog

题目描述

某地有 N 个能量发射站排成一行，每个发射站 i 都有不相同的高度 Hi，并能向两边（当 然两端的只能向一边）同时发射能量值为 Vi 的能量，并且发出的能量只被两边最近的且比 它高的发射站接收。

显然，每个发射站发来的能量有可能被 0 或 1 或 2 个其他发射站所接受，特别是为了安 全，每个发射站接收到的能量总和是我们很关心的问题。由于数据很多，现只需要你帮忙计 算出接收最多能量的发射站接收的能量是多少。

输入输出格式

第 1 行：一个整数 N;

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

输出仅一行，表示接收最多能量的发射站接收到的能量值，答案不超过 longint。

输入输出样例

3
4 2 
3 5 
6 10

7

说明

对于 40%的数据，1<=N<=5000；1<=Hi<=100000；1<=Vi<=10000;

对于 70%的数据，1<=N<=100000；1<=Hi<=2,000,000,000；1<=Vi<=10000;

对于 100%的数据，1<=N<=1000000;1<=Hi<=2,000,000,000；1<=Vi<=10000。

简单的单调栈裸题。考虑到一个点的接收的能量不是从前面来，就是从后面来。因此我们用单调栈边弹栈边统计栈内获得的能量和当前位置的能量就能O（n）得到答案了（另外pair真好用）

#include<cstdio>
#include<algorithm>
#include<stack>
#define STKTOP (stk.empty() ? 0 : stk.top().first)
using namespace std;
int n, h, tp, mx;
int val[1000007];
int v[1000007];
typedef pair<int, int> pii;
stack<pii > stk;
int main() {
    scanf("%d", &n);
    for(register int i = 1; i <= n; ++i) {
        scanf("%d%d", &h, &v[i]);
        while(!stk.empty() && stk.top().first <= h) 
            val[i] += v[stk.top().second], stk.pop();
        if(!stk.empty())val[stk.top().second] += v[i];
        stk.push(make_pair(h, i));
    }
    for(register int i = 1; i <= n; ++i) {
        mx = max(val[i], mx);
    }
    printf("%d", mx);
    return 0;
}