链接:https://ac.nowcoder.com/acm/problem/14666
来源:牛客网
题目描述
M国的地势高低不平,现给出一个数组代表此国家某纬度上均匀分布的N座山的海拔高度Hi,已知每座山的山顶上都有一座哨塔,若两个哨兵分别位于第i、j(i<j)座山上,当且仅当两人所在的山比两人之间所有的山都高时,这两个哨兵可以相互监视,M国的防守能力大小为相互监视的哨兵对数。H国早已对M国虎视眈眈,H国的皇帝希望黑魔法师们可以在M国的某两座山之间放置一块巨大的屏障,M国的哨兵不可通过该屏障互相监视。皇帝想让你告诉他最优的屏障放置位置,你是皇帝手下最信任的军师,现在需要你帮助皇帝计算最优的屏障放置位置和最大的防守力减少量。
输入描述:
第一行包含一个正整数T(T≤20)。
对于每组数据,第一行包含一个正整数n(2≤n≤50000)。
接下来n个不同的正整数,H1,H2,H3,…,Hn(0≤Hi≤109)分别代表横截面上每座山的海拔高度。
(读入数据比较大,建议使用scanf而不要使用cin读入)
对于60%的数据,n≤500
对于80%的数据,n≤5000
对于100%的数据,n≤50000
输出描述:
每组数据输出一行形如“Case #N: X C”,N代表当前是第N组数据(从1开始),X代表屏障放置在第X座山前可使M国的防守能力下降最多, 此时减少量为C。若有多种方案使得减少量为C,那么输出最小的X对应的方案。
输入
2
3
2 1 3
5
4 5 2 6 3
输出
Case #1: 2 2
Case #2: 3 2
本题的理解
先分析题目,现在判断一个山的防御值可以向前遍历直到找到比它高的山或尽头,也可以向后遍历直到找到比它高的山或尽头,这一单调递增的特性可以用入栈、弹栈的操作进行,找出每一个向前的防御值以及向后的防御值,而程序中用到的思想是前缀和的思想,这样只需遍历一次,也就是每次向前遍历时先加上上一个山的防御值。
其实最重要的是理解maxnn=vis[n]-(vis[i]+v[i+1]); 该语句先是vis[n]-vis[i]=v[i] (这里是便于理解但是实际上上不等) v[i]-v[i+1]则表示向后寻找的山的个数而不算初始值。
#include<iostream>
#include<cstring>
#include<stack>
using namespace std;
const int maxn=50005;
int a[maxn],v[maxn],vis[maxn];
int main(){
//freopen("qwe.txt","r",stdin);
int T,cs,n;
scanf("%d",&T);
stack<int > q;
while(T--)
{
cs++;
memset(v,0,sizeof(v));
memset(vis,0,sizeof(vis));
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
//“正向遍历判断”,即从头到尾然后利用栈的特性,记录每点(山)向前看过去
// 一直到看到比当前山高的山或找到了了尽头
while(!q.empty()) q.pop();//每次使用栈时都要进行清空的操作
for(int i=1;i<=n;i++)
{
int tmp=0;
vis[i]=vis[i-1];//每个山的初始值都设为前一个山的防御值
//这样vis数组存储的就是前i个山的防御值的总和
while(!q.empty()&&q.top()<a[i])
{
//向前遍历直到高山或尽头
q.pop();
tmp++;
}
if(!q.empty())
vis[i]+=tmp+1;//栈中有山即表示找到的是高山防御值要加一
else vis[i]+=tmp;//找到的是尽头
//这样最终每个vis[i]存储的就是前i个山的防御值
q.push(a[i]);
}
while(!q.empty()) q.pop();
//下面是“逆向遍历判断”,即从尾到头的遍历,记录的是从当前山向后看过去
//一直到比当前高的山或尽头
for(int i=n;i>=1;i--)
{
int tmp=0;
v[i]=v[i+1];//同样每个山的初始值都为后一个山的防御值
//这样v[i]表示的就是从i个山到最后的山的总防御和
while(!q.empty()&&q.top()<a[i])
{
tmp++;
q.pop();
}
if(!q.empty())v[i]+=tmp+1;
else v[i]+=tmp;
q.push(a[i]);
}
int maxnn=-1,id=0;
for(int i=1;i<n;i++)
{
if(v[1]-(vis[i]+v[i+1])>maxnn)
{
//这一步判断可以这样理解
// 首先vis[n]和v[1]的意义是一样的都是总防御值
//下面语句可以这样理解vis[n]-vis[i]=v[i]的即总防御值先去i山的正向遍历防御
//和等于当前山的逆向遍历防御和
//=v[i]-v[i+1]即为i山的去除初始值的防御和
maxnn=vis[n]-(vis[i]+v[i+1]);
id=i;
}
}
printf("Case #%d: %d %d\n",cs,id+1,maxnn);
}
return 0;
}