POJ 3250 Bad Hair Day

 

链接：http://poj.org/problem?id=3250

题意：N头牛面向右排成一排，每头牛可以看到它前面的比它身高严格矮的牛。给出每头牛的身高，求出所有牛能看到的牛的总数。

分析：单调队列。牛i能看到的牛的数量是j-i-1（设j>i，且牛j是第一个身高大于或等于牛i的牛）。所以每当入队一头牛，就计算它后面比它矮的牛能看到的牛的数量，并将该牛T掉（因为它能看到的牛的数量已经计算出来了，它也不会影响到其前面或后面的牛的计算）。最后在队尾插入一个正无穷的数，将队中剩下的牛都计算出来。

 

Source Code：

#include<stdio.h>
#define INF 0x3f3f3f3f
#define maxn 80010

struct node{
	int val,idx;
}que[maxn];
int height[maxn];

int main()
{
	int n,i,x,head,tail;
	__int64 ans=0;
	scanf("%d",&n);
	for(i=0;i<n;i++)
		scanf("%d",&height[i]);
	height[i]=INF;
	head=tail=0;
	for(i=0;i<=n;i++){
		while(head<tail&&height[i]>=que[tail-1].val){
			ans+=i-que[tail-1].idx-1;
			tail--;
		}
		que[tail].val=height[i];
		que[tail++].idx=i;
	}
	printf("%I64d\n",ans);
}