P2947 [USACO09MAR] Look Up S

#include<iostream>
#include<stack>
using namespace std;
int h[100005],ans[100005];
//h数组储存每个奶牛的高度
//ans数组储存比i个奶牛高的那个奶牛是第几个 
int main()
{
	int n;
	cin>>n;
	stack<int>cow;
	for(int i=1;i<=n;i++)
	{
		cin>>h[i];
	}
	for(int i=n;i>=1;i--)
	{
		while(!cow.empty()&&h[cow.top()]<=h[i])//判断此奶牛是否比栈顶高 
		{
			cow.pop();//弹出比新奶牛矮的栈顶 
		}
		//while循环最后的栈顶比新奶牛高，或者栈空了没有比它高的 
		if(cow.empty()) ans[i]=0;//如果栈空了，那么没有比它高的 
		else ans[i]=cow.top();//栈没空，就输出此时的栈顶 
		cow.push(i);//将此奶牛入栈 
	}
	for(int i=1;i<=n;i++)
	{
		cout<<ans[i]<<endl;
	} 
	return 0;
}