数据结构 -- 单调栈

1、题目描述

给定一个长度为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

 

2、分析

（1）栈的数据结构：

 

（2）性质：

• 对于一个栈中的元素，若存在 stk[i] >= stk[j] && i < j，那么可以推出：stk[i]不可能出现在答案中，因此，stk[i]可以被删掉；于是，删去像这样的所有的stk[i]，栈就变成了一个严格单调递增的序列。
• 当我们想找到栈中第k个数的左边第一个比它小的数；也就是栈初始化为空，我们每次往栈中插入一个新的元素，记为num，

需要先与栈顶元素比较：

若stk[tt] >= num，由于新的元素num在当前栈顶shan'c的右边插入，所以当前栈顶元素stk[tt]永远不可能当成答案被输出，所以当前栈顶可以删除。

并且从栈顶至栈底的顺序比较，直到找到第一个元素stk[tt] < num为止，将num插入到栈中。

 

3、代码

（1）朴素写法

import java.io.*;
import java.util.*;

public class Main{
    static int N = 100010;
    static int[] nums = new int[N];
    
    public static void main(String[] args){
        Scanner in = new Scanner(new InputStreamReader(System.in));
        int n = in.nextInt();
        
        for(int i = 0;i < n;i ++){
            nums[i] = in.nextInt();
        }
        
        System.out.print(-1 + " ");
        
        int j;
        for(int i = 1;i < n;i ++){
            for(j = i-1;j >= 0;j --){ //j从i-1往左移动寻找
                if(nums[i] > nums[j] && j >= 0){
                    System.out.print(nums[j] + " ");
                    break;
                }
            }
            if(j <0){
                System.out.print(-1 + " ");
            }
        }
    }
    
}

 

（2）正确写法

import java.io.*;
import java.util.*;

public class Main{
    static int N = 100010;
    static int[] stk = new int[N]; //栈
    static int tt = 0; //栈顶
    
    public static void main(String[] args){
        Scanner in = new Scanner(new InputStreamReader(System.in));
        int n = in.nextInt();
        
        for(int i = 0;i < n;i ++){
            int x = in.nextInt();
            while(tt > 0 && stk[tt] >= x){
                tt --;
            }
            if(tt > 0){
                System.out.print(stk[tt] + " ");
            }else{
                System.out.print(-1 + " ");
            }
            stk[++ tt] = x;
        }
        
    }
    
}