题目链接:链接: NK—14666.
题目描述:
M国的地势高低不平,现给出一个数组代表此国家某纬度上均匀分布的N座山的海拔高度Hi,已知每座山的山顶上都有一座哨塔,若两个哨兵分别位于第i、j(i<j)座山上,当且仅当两人所在的山比两人之间所有的山都高时,这两个哨兵可以相互监视,M国的防守能力大小为相互监视的哨兵对数。H国早已对M国虎视眈眈,H国的皇帝希望黑魔法师们可以在M国的某两座山之间放置一块巨大的屏障,M国的哨兵不可通过该屏障互相监视。皇帝想让你告诉他最优的屏障放置位置,你是皇帝手下最信任的军师,现在需要你帮助皇帝计算最优的屏障放置位置和最大的防守力减少量。
Input:
第一行包含一个正整数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
Output:
每组数据输出一行形如“Case #N: X C”,N代表当前是第N组数据(从1开始),X代表屏障放置在第X座山前可使M国的防守能力下降最多, 此时减少量为C。若有多种方案使得减少量为C,那么输出最小的X对应的方案。
Example:
input:
2
3
2 1 3
5
4 5 2 6 3
output:
Case #1: 2 2
Case #2: 3 2
思路:插入屏障后,区间被分为[1,x]和[x+1,n]两个部分。可以利用栈来求出前缀的对数和、后缀的对数和,前缀和表示连续下降的山的个数,后缀和同理。然后在一个for循环来寻找最大值,减少的最大防守力就是[1,n] - [1,x] - [x+1,n],最优的屏障放置位置就是当前的x+1。
#include<bits/stdc++.h>
#include<iostream>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
#define IOS std::ios::sync_with_stdio(false)
#define ls p<<1
#define rs p<<1|1
#define mod 1000000000 + 7
#define PI acos(-1.0)
#define INF 1e9
#define N 200000 + 5
/*********************Code*********************/
int t,n,ans,pos,sum,vis1[N],vis2[N],a[N];
stack<int>st;
int main(void){
scanf("%d",&t);
for(int k = 1;k <=t;k++){
cin>>n;
vis1[0] = vis2[n+1] = ans = pos = 0;
for(int i = 1;i <=n;i++)
scanf("%d",&a[i]);
while(st.size())
st.pop();
for(int i = 1;i <=n;i++){
vis1[i] = vis1[i-1];
sum = 0;
while(!st.empty()&&st.top()<a[i]){
sum++;
st.pop();
}
if(!st.empty())
vis1[i] +=sum+1;
else
vis1[i] +=sum;
st.push(a[i]);
}
while(!st.empty())
st.pop();
for(int i = n;i>=1;i--){
vis2[i] = vis2[i+1];
sum = 0;
while(!st.empty()&&st.top()<a[i]){
sum++;
st.pop();
}
if(!st.empty())
vis2[i] +=sum+1;
else
vis2[i] +=sum;
st.push(a[i]);
}
for(int i = 1;i<=n;i++){
if(vis1[n]-vis1[i]-vis2[i+1]>ans){
ans = vis1[n] - vis1[i] -vis2[i+1];
pos = i + 1;
}
}
printf("Case #%d: %d %d\n",k,pos,ans);
}
return 0;
}