【洛谷P2947】向右看齐

向右看齐

题目链接

此题可用单调栈O(n)求解

维护一个单调递减栈，元素从左到右入栈

若新加元素大于栈中元素，则栈中元素的仰望对象即为新加元素

每次将小于新加元素的栈中元素弹出，记录下答案

#include<iostream>
#include<cstdio>
using namespace std;
#define N 100010
int n,w[N],stack[N],top,ans[N];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
     scanf("%d",&w[i]);
    w[0]=1000100;
    for(int i=1;i<=n;i++){
        while(w[stack[top]]<w[i]&&top>0){
            ans[stack[top]]=i;
            top--;
        }
        stack[++top]=i;
    }
    for(int i=1;i<=n;i++)
     printf("%d\n",ans[i]);
    return 0;
}