每个数左边第一个比它小/大的数(单调栈)

给定一个长度为N的整数数列,输出每个数左边第一个比它小的数,如果不存在则输出-1。

输入格式

第一行包含整数N,表示数列长度。

第二行包含N个整数,表示整数数列。

输出格式

共一行,包含N个整数,其中第i个数表示第i个数的左边第一个比它小的数,如果不存在则输出-1。

数据范围

1≤N≤1051≤N≤105
1≤数列中元素≤1091≤数列中元素≤109

输入样例:

5
3 4 2 7 5

输出样例:

-1 3 -1 2 2

解题思路:

利用栈的先进后出思想,来维持栈的单调性

拿测试案例来作为例子

首先3的左边没有数字了,因此输出-1,将3入栈

然后4的左边只有3且比4更加小,那么就输出3,将4入栈,此时栈中的数字是3和4

3 和 4 比 2 都要大,将2压入栈中会破坏栈的单调性,那么就将比2都大的3和4都弹出,将2压入栈中并输出-1

7比2要大,压入栈中不会破坏单调性,同时7左边最小的数字是2,因此将7压入栈中后输出2

将5压入栈中会破坏单调性,将7弹出,输出2,将5压入栈中,算法结束

#include 
using namespace std;
const int N = 100010;
int stk[N],a[N],tt = 0;
int n;
int main(){
    cin >> n;
    for(int i = 0; i < n; ++i) scanf("%d",&a[i]);
    for(int i = 0; i < n; ++i){
        while(tt && stk[tt] >= a[i]) tt--;
        if(tt) printf("%d ",stk[tt]);
        else printf("-1 ");
        
        stk[++tt] = a[i];
    }
    return 0;
}

 

 

 

你可能感兴趣的:(leetcode刷题)