【题解】P2947

P2947

题目

题目描述：

约翰的 $N(1\le N\le10^5)$ 头奶牛站成一排，奶牛 $i$ 的身高是 $H_i(1\le H_i\le10^6)$。现在，每只奶牛都在向右看。对于奶牛 $i$，如果奶牛 $j$ 满足 $i<j$ 且 $H_i<H_j$，我们可以说奶牛 $i$ 可以仰望奶牛 $j$。 求出每只奶牛离她最近的仰望对象。

输入格式：

第 $1$ 行输入 $N$，之后 $N$ 行第 $i+1$ 行输入一个身高 $H_i$。

输出格式：

共 $N$ 行，按顺序每行输出一只奶牛的最近仰望对象，如果没有仰望对象，输出 $0$。

数据范围与说明：

【输入说明】

$6$ 头奶牛的身高分别为 $3$, $2$, $6$, $1$, $1$, $2$。

【输出说明】

奶牛 $1,2$ 仰望奶牛 $3$，奶牛 $4,5$ 仰望奶牛 $6$，奶牛 $3$ 和 $6$ 没有仰望对象。

【数据规模】

对于 $20\%$ 的数据：$1\le N\le10$；

对于 $50\%$ 的数据：$1\le N\le10^3$；

对于 $100\%$ 的数据：$1\le N\le10^5,1\le H_i\le10^6$。

输入输出样例 #1

输入：

6 
3 
2 
6 
1 
1 
2

输出：

3 
3 
0 
6 
6 
0

代码

#include<bits/stdc++.h>
using namespace std;
int h[100001],ans[100001];
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&h[i]);
    }
    stack<int> st;
    for(int i=n;i>=1;i--){
        while(!st.empty()&&h[st.top()]<=h[i]){
            st.pop();
        }
        if(st.empty()){
            ans[i]=0;
        }else{
            ans[i]=st.top();
        }
        st.push(i);
    }
    for(int i=1;i<=n;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}