【洛谷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=;i<=n;i++)
      scanf("%d",&w[i]);
     w[]=;
     for(int i=;i<=n;i++){
         while(w[stack[top]]<w[i]&&top>){
             ans[stack[top]]=i;
             top--;
         }
         stack[++top]=i;
     }
     for(int i=;i<=n;i++)
      printf("%d\n",ans[i]);
     return ;
 }