【單調棧】向右看齊Look Up

向右看齊Look Up

題目

約翰的N(1≤N≤10^5)頭奶牛站成一排，奶牛i的身高是Hi(l≤Hi≤1,000,000)．現在，每隻奶牛都在向右看齊．對於奶牛i，如果奶牛j滿足i<j且Hi<Hj，我們可以說奶牛i可以仰望奶牛j． 求出每隻奶牛離她最近的仰望對象．

輸入

第 1 行輸入 N，之後每行輸入一個身高 H_i。

輸出

共 N 行，按順序每行輸出一隻奶牛的最近仰望對象，如果沒有仰望對象，輸出 0。

輸入樣例

6 
3 
2 
6 
1 
1 
2 

輸出樣例

3 
3 
0 
6 
6 
0 

說明

【輸入說明】6 頭奶牛的身高分別爲 3, 2, 6, 1, 1, 2.

【輸出說明】奶牛#1,#2 仰望奶牛#3，奶牛#4,#5 仰望奶牛#6，奶牛#3 和#6 沒有仰望對象。

數據範圍

對於 20%的數據： 1≤N≤10；

對於 50%的數據： 1≤N≤1,000；

對於 100%的數據：1≤N≤100,000；1≤H_i≤1,000,000；

解題思路

其實這道題就是一道單調棧，先把第一個奶牛給扔進去，然後再判斷是否有比它大的，如果有大的就出棧，比它小的就把它指向棧裏的奶牛然後重複執行即可.

程序如下

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int n,a[100001],ans[100001],sum;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	   scanf("%d",&a[i]);
	for(int i=n-1;i>=1;--i)
	{
		sum=i+1;
		while(a[i]>=a[sum]&&a[sum]>0) sum=ans[sum];//判斷是否有比它大的
		ans[i]=sum;//有就替換									
	} 
	for(int i=1;i<=n;++i) printf("%d\n",ans[i]);
	return 0;
}