腾讯2020校园招聘-后台 第二题

题目：

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

小Q从第一栋一直走到了最后一栋，小Q从来都没有见到这么多的楼，

所以他想知道他在每栋楼的位置处能看到多少栋楼呢？（当前面的楼的高度大于等于后面的楼时，后面的楼将被挡住） 

输入例子1:

[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栋楼。

 

代码：

//暴力法超时，未通过

/**
 * @author zcy
 * @date 2021年04月08日 12:46
 */
public class Solution {

    public int[] findBuilding (int[] heights) {
        // write code here
        int left=0, right=0;
        int count=0;
        int[] res=new int[heights.length];
        for (int i = 0; i < heights.length; i++) {
            //i为当前统计位置，count为统计值，再最后要加上自己（能看到自己）
            left=i-1;
            right=i+1;
            count=0;
            //统计前方（左侧）可看
            int hight=0;    //遍历后方时，得到的当前最高的房屋高度，比此低和同高，看不到
            while(left>=0){
                if(hight<heights[left]){
                    count++;
                    hight=heights[left];
                }
                left--;
            }
            hight=0;

            //统计后方（右侧）可看
            while(right<heights.length){
                if(hight<heights[right]){
                    count++;
                    hight=heights[right];
                }
                right++;
            }
            res[i]=++count;
        }
        return res;
    }
}

 

代码：

package com.zcy.mybatisplus.tx;/**
 * @author chenyang zhang
 * @date 2021/4/8 - 12:46
 */

import java.util.Arrays;
import java.util.Stack;

/**
 * @author zcy
 * @date 2021年04月08日 12:46
 */
//采用单调递减栈，栈顶元素最小，因为前面房屋矮，不会挡住后面；
// 前面高，才会挡住后面，栈顶是最接近当前位置房屋，所以要最小，
// 代表此时栈顶所在位置向某一方向能看到房屋。
public class Solution {

    public int[] findBuilding (int[] heights) {
        // write code here
        Stack<Integer> stack=new Stack<>();
        int len=heights.length;
        int[] res=new int[len];
        //至少可以看到自己
        Arrays.fill(res,1);
        //从后往前看，如果当前位置比栈顶小，那么可以看到
        for (int i = len-1; i >=0 ; i--) {
            res[i]+=stack.size();
            //此时是为了保证单调递减，求的下一位置可看到的房屋数量
            while(!stack.isEmpty()&&stack.peek()<=heights[i]){
                stack.pop();
            }
            stack.push(heights[i]);
        }
        //将栈恢复到空栈，计算下一个方向
        stack.clear();

        //从前往后看，如果当前位置比栈顶小，那么站在其后面一个位置可以看到当前房屋
        //反之当前房屋比栈顶高，站在后面一个位置就看不到此时的栈顶这个房屋，所以要退栈
        for (int i = 0; i <=len-1 ; i++) {
            res[i]+=stack.size(); //记录当前房屋朝后看额能看到的数量
            while(!stack.isEmpty()&&stack.peek()<=heights[i]){
                stack.pop();
            }
            stack.push(heights[i]);
        }
        return res;
    }

    public static void main(String[] args) {
        int[] arr= {5, 3, 8, 3, 2, 5};
        Solution solution=new Solution();
        int[] res=solution.findBuilding(arr);
    }
}