POJ 3250 Bad Hair Day 单调栈 代码详解

欢迎关注我的个人博客：www.zuzhiang.cn

 

传送门：POJ 3250

 

题目大意：有一群牛站成一排，每头牛都是面朝右的，每头牛可以看到他右边身高比他小的牛。给出每头牛的身高，要求每头牛能看到的牛的总数。

本题还有一个类似的题，那就是给出N块竖直的木板的高度，现在在第 i 块木板的右方加水，问可以水可以淹没多少木板。

 

 

Sample Input

6
10
3
7
4
12
2

Sample Output

5

 

前置技能：单调栈的原理及应用。

 

 

思路：也就是说针对每只牛，啊，不，每头牛求它到它右边第一个比它高（或身高相等）的牛之间有多少头牛，然后将求得的结果相加就是最后的答案。朴素的做法是针对每头牛去寻找右边比它高的牛的位置，时间复杂度为O(n^2)，如果用单调栈的话就是O(n).

 

具体实现：利用根据单调递增栈解决，如果栈为空或入栈元素小于栈顶元素则入栈，否则会破坏栈的单调性，则将栈顶元素出栈并更新结果，直到栈为空或碰到一个小于入栈元素的值。然后将当前元素入栈。

设数组的最后一个元素为最大值，也就相当于在最右边的牛的右边设了一个高度无限高的牛。这样做的目的是，最后让栈内的元素全部出栈。

 

PS：我的代码中的单调栈保存的是牛的位置。

 

PSS：结果应该用long long 型，最多有80000头牛，每头牛右边最多有80000头牛，80000*80000=6.4*10^9，而int最多表示大约为2*10^9……之前给想错了……哭死……

 

 

//为什么会超过int型…… 
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<stack>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long LL;

int main()
{
	int i,n,top,a[80010]; //top指向栈顶元素 
	LL ans; //存储最终结果 
	stack<int> st; //st为栈，每次入栈的是每头牛的位置 
	while(~scanf("%d",&n))
	{
		while(!st.empty()) st.pop(); //清空栈 
		for(i=0;i<n;i++)
			scanf("%d",&a[i]);
		a[n]=inf; //将最后一个元素设为最大值 
		ans=0;
		for(i=0;i<=n;i++)
		{
			if(st.empty()||a[i]<a[st.top()])
			{ //如果栈为空或入栈元素小于栈顶元素，则入栈 
				st.push(i);
			}
			else 
			{
				while(!st.empty()&&a[i]>=a[st.top()])
				{ //如果栈不为空并且栈顶元素不大于入栈元素，则将栈顶元素出栈 
					top=st.top(); //获取栈顶元素 
					st.pop();     //栈顶元素出栈 
					//这时候也就找到了第一个比栈顶元素大的元素 
					//计算这之间牛的个数，为下标之差-1 
					ans+=(i-top-1);	
				}
				st.push(i); //当所有破坏栈的单调性的元素都出栈后，将当前元素入栈 
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}