腾讯2020校招——小Q逛街看楼

题目

小Q在周末的时候和他的小伙伴来到大城市逛街，一条步行街上有很多高楼，共有n座高楼排成一行。

小Q从第一栋一直走到了最后一栋，小Q从来都没有见到这么多的楼，所以他想知道他在每栋楼的位置处能看到多少栋楼呢？（当前面的楼的高度大于等于后面的楼时，后面的楼将被挡住） 

输入描述:

输入第一行将包含一个数字n，代表楼的栋数，接下来的一行将包含n个数字wi(1<=i<=n)，代表每一栋楼的高度。

1<=n<=100000;

1<=wi<=100000; 

输出描述:

输出一行，包含空格分割的n个数字vi，分别代表小Q在第i栋楼时能看到的楼的数量。

输入例子1:

6
5 3 8 3 2 5

输出例子1:

3 3 5 4 4 4

例子说明1:

当小Q处于位置3时，他可以向前看到位置2,1处的楼，向后看到位置4,6处的楼，加上第3栋楼，共可看到5栋楼。当小Q处于位置4时，他可以向前看到位置3处的楼，向后看到位置5,6处的楼，加上第4栋楼，共可看到4栋楼。

解题

方法1：暴力法

public static void Num1(){
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        int a[] = new int[N];
        for(int i = 0;i < N;i++){
            a[i] = in.nextInt();
        }
        for(int i =0;i<N;i++){
            int maxLeft = -1,maxRight = -1;
            int num = 1;
            // 看左边的
            for(int j = i-1;j>=0;j--){
                if(a[j]>maxLeft){
                    maxLeft = a[j];
                    num++;
                }
            }
            // 看右边的
            for(int j = i+1;j<N;j++){
                if(a[j]>maxRight){
                    maxRight = a[j];
                    num++;
                }
            }
            System.out.print(num+" ");
        }
    }

此种方法虽然可以解，但是复杂度很高，并不满足题目要求

方法2：单调栈方法

public static void Num(){
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        int a[] = new int[N];
        for(int i = 0;i < N;i++){
            a[i] = in.nextInt();
        }
        int b[] = new int[N];
        // 单调栈
        Stack<Integer> Lstack = new Stack<>();
        Stack<Integer> Rstack = new Stack<>();

        for (int i = 0;i<N;i++){
            // 向左看，加上元素个数
            b[i] += Lstack.size();
            // 维护左栈，如果后面加的大于栈顶的元素，就弹出
            while (!Lstack.empty() && a[i]>=Lstack.peek()) {
                Lstack.pop();
            }
            Lstack.add(a[i]);
        }

        for (int i = N-1;i>=0;i--){
            // 向右看，加上元素个数
            b[i] += Rstack.size();
            // 维护右栈，从后往前，如果大于栈顶元素，弹出
            while (!Rstack.empty() && a[i]>=Rstack.peek()){ 
                Rstack.pop();
            }
            Rstack.add(a[i]);
        }

        for (int i = 0;i<N;i++){
            System.out.print(b[i]+1 + " ");
        }
    }