JZOJ_1264_乱头发节

Description

　　N 头奶牛 (1 <= N <= 80,000) 正在过乱头发节！由于每头牛都意识到自己凌乱不堪的发型，FJ 希望统计出能够看到其他牛的头发的牛的数量。 　　每一头牛 i有一个高度 h[i] (1 <= h[i] <= 1,000,000,000)而且面向东方排成一排(在我们的图中是向右)。因此，第i头牛可以看到她前面的那些牛的头，（即i+1, i+2,等等），只要那些牛的高度严格小于她的高度。
       每一头牛 i有一个高度 h[i] (1 <= h[i] <= 1,000,000,000)而且面向东方排成一排(在我们的图中是向右)。因此，第i头牛可以看到她前面的那些牛的头，（即i+1, i+2,等等），只要那些牛的高度严格小于她的高度。
例如这个例子:

              =
=            =
=            =
=    =      =           牛面向右侧 -->
=    =      =
= = = =  =
= = = =  = =
1 2 3 4  5 6


牛#1 可以看到她们的发型 #2, 3, 4
牛#2 不能看到任何牛的发型
牛#3 可以看到她的发型 #4
牛#4 不能看到任何牛的发型
牛#5 可以看到她的发型 6
牛#6 不能看到任何牛的发型!
让 c[i] 表示第i头牛可以看到发型的牛的数量；请输出 c[1] 至 c[N]的和。如上面的这个例子，正确解是3 + 0 + 1 + 0 + 1 + 0 = 5。

Input

• Line 1: 牛的数量 N。
• Lines 2..N+1: 第 i+1 是一个整数，表示第i头牛的高度。
   

Output

• Line 1: 一个整数表示c[1] 至 c[N]的和。

Sample Input

6
10
3
7
4
12
2

Sample Output

5

 

思路

我们可以维护一个单调递减的栈，每次栈的长度减去一位（首位）的剩余长度的总和就是每头牛可以看见的数量和

如样例数据：

10 3 7 4 12 2

每一次栈的状态：

1. 栈：10        ===>长度：1  ；ans=1-1=0（长度减1，下同）
2. 栈：10 3     ===>长度：2  ；ans=0+2-1=1
3. 栈：10 7     ===>长度：2  ；ans=1+2-1=2
4. 栈：10 7 4  ===>长度：3  ；ans=2+3-1=4
5. 栈：12        ===>长度：1  ；ans=4+1-1=4
6. 栈：12 2     ===>长度：2  ；ans=4+2-1=5

所以最终答案就是5

因为每次栈中的数（除第一位外）都是可以被后面的牛看见的牛的高度，每次单调递减就可以保证每一个都可以被后面的看见，而如果后来要入栈一个超大的数，而他就会挡住前面的牛，所以就要一直出栈直到栈尾的数比要入的数要大。

#include<iostream> 
#include<cstdio> 
#include<algorithm> 
#include<cmath>
#define maxn 80005
using namespace std;

int zz[maxn],i,j,n,m,tot;
long long ans;

int main()
{
	freopen("badhair.in","r",stdin);
	freopen("badhair.out","w",stdout);
	
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&m);
		if (m<zz[tot]||tot==0) zz[++tot]=m;//如果栈尾已经比m大，那么直接入栈
		else 
		{
			while (zz[tot]<=m&&tot>0) //出栈直到栈尾比m大
			{
			    zz[tot]=0;
				tot--;
		    }
			zz[++tot]=m;//将m入栈
		}
		ans=ans+max(tot-1,0);//统计ans
	}
	printf("%lld",ans);
}