acwing 600. 仰视奶牛

题目地址  https://www.acwing.com/problem/content/description/602/

约翰有N头奶牛，编号为1到N。

现在这N头奶牛按编号从小到大的顺序站成了一排，其中奶牛 i 的身高为HiHi。

现在，每头奶牛都向它的右侧望向那些编号较大的奶牛，对于奶牛 i 如果存在一头奶牛 j 满足 i<ji<j 并且 Hi<HjHi<Hj，那么我们称奶牛 i 需要仰视奶牛 j。

请你求出每头奶牛的最近仰视对象。

输入格式

第一行包含整数N。

接下来N行，每行包含一个整数HiHi，其中第 i 行的数为编号为 i 的奶牛的高度。

输出格式

共 N 行，每行输出一个整数，其中第 i 行的输出整数表示编号为 i 的奶牛的最近仰视对象的编号，如果不存在仰视对象，则输出0。

数据范围

1≤N≤1e5
1≤Hi≤1e6

 

输入样例：
6 
3 
2 
6 
1 
1 
2 
输出样例：
3 
3 
0 
6 
6 
0 

 

 

单调栈记录最近的未仰视的牛编号(有仰视的答案已经记录了 出栈)

#include <iostream>
#include <vector>
#include <stack>


using namespace std;

const int N = 1e5+100;

vector<int> arr(N,0);
vector<int> ans(N,0);
stack<int> st;
int n ;

int main()
{
    cin >> n;
    for(int i =1;i<= n;i++){
        cin >> arr[i];
    }
    
    for(int i =1;i <= n;i++){
        while(!st.empty() && arr[i]>arr[st.top()]){
            ans[st.top()] = i;
            st.pop();
        }
        
        st.push(i);
    }
    
    while(!st.empty()){
         ans[st.top()] = 0;
         st.pop();
    }
    
    for(int i = 1 ;i<=n;i++){
        cout << ans[i] << endl;
    }
    
    
    return 0;
}