最优屏障（栈）

最优屏障

题目：

题目描述：
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对应的方案。

输入：

2
3
2 1 3
5
4 5 2 6 3

输出:

Case #1: 2 2
Case #2: 3 2

思路：分别开前缀和后缀两个数组，每次用栈来维护数组内储存的哨兵对数.

最后用 ：前缀[n]-( 前缀[i-1]+后缀[i] )遍历模拟答案；

来看一下代码：

#include<bits/stdc++.h>
using namespace std;
int a[101010],q[101010],h[101010];
int main()
{
	int t;
	cin>>t;
	for(int k=1;k<=t;k++)
	{
		memset(q,0,sizeof(q));//多实例，每次都要初始化
		memset(h,0,sizeof(h));
		int n;
		cin>>n;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
		}
		stack<int> st;
		for(int i=1;i<=n;i++)//前缀
		{
			q[i]=q[i-1];//计算往前看的最大防御力前缀和
			int ans=0;
			while(!st.empty()&&st.top()<a[i])//遇到比自己高的就开始清栈
			{
				ans++;
				st.pop();
			}
			if(!st.empty())
			q[i]+=ans+1;//把当前最高的也加上
			else
			q[i]+=ans;
			st.push(a[i]);
		}
		while(!st.empty())//先清栈，再计算后缀
		{
			st.pop();
		}
		for(int i=n;i>0;i--)//后缀
		{
			h[i]=h[i+1];
			int ans=0;
			while(!st.empty()&&st.top()<a[i])
			{
				ans++;
				st.pop();
			}
			if(!st.empty())
			h[i]+=ans+1;
			else
			h[i]+=ans;
			st.push(a[i]);
		}
		int maxx=0,j=0;
		for(int i=1;i<=n;i++)
		{
			if(q[n]-(q[i-1]+h[i])>maxx)
			{
				maxx=q[n]-(q[i-1]+h[i]); //遍历找最大
				j=i;
			}
		}
		cout<<"Case #"<<k<<": "<<j<<" "<<maxx<<endl;
	}
	return 0;
}

下一篇 MoonLight的运算问题