洛谷 P1901 发射站

题目描述

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

请计算出接收最多能量的发射站接收的能量是多少。

输入格式

第 1 行一个整数 N。

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

输出格式

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

输入输出样例

输入 #1复制

3 4 2 3 5 6 10

输出 #1复制

7

说明/提示

对于 40% 的数据，1≤N≤5000,1≤Hi​≤105,1≤Vi​≤104。

对于 70% 的数据，1≤N≤105,1≤Hi​≤2×109,1≤Vi​≤104。

对于 100% 的数据，1≤N≤106,1≤Hi​≤2×109,1≤Vi​≤104。

#include<bits/stdc++.h> using namespace std; const int N=1e6+10; int h[N],v[N]; int n; int sum[N]; int main() { cin>>n; for(int i=1;i<=n;i++) { cin>>h[i]>>v[i]; } //找左边 stack<int> st; for(int i=1;i<=n;i++) { while(st.size()&&h[st.top()]<=h[i]) st.pop(); if(st.size()) { sum[st.top()]+=v[i]; } st.push(i); } //清空栈内元素 while(st.size()) st.pop(); //找右边 for(int i=n;i>=1;i--) { while(st.size()&&h[st.top()]<=h[i]) st.pop(); if(st.size()) { sum[st.top()]+=v[i]; } st.push(i); } int ret=0; for(int i=1;i<=n;i++) { ret=max(ret,sum[i]); } cout<<ret<<endl; return 0; }