原问题: 看到多少栋楼?
小Q在周末的时候和他的小伙伴来到大城市逛街,一条步行街上有很多高楼,共有n座高楼排成一行。
小Q从第一栋一直走到了最后一栋,小Q从来都没有见到这么多的楼,所以他想知道他在每栋楼的位置处能看到多少栋楼呢?(当前面的楼的高度大于等于后面的楼时,后面的楼将被挡住)
解法:用单调栈
package com.company;
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[] arr = new int[n];
int[] ans = new int[n];
for(int i = 0; i < n; i++){
arr[i]=in.nextInt();
}
Stack<Integer> L = new Stack<>();
for (int i = 0; i < n;i++){
ans[i] = L.size();//保存的是i往左看,能看到的全部楼高和
if (i != 0) {
while (!L.empty() && arr[i] >= L.peek()) {//如果这个楼能把它前面的楼挡住,那就把被挡住的楼移除
L.pop();
}
}
L.push(arr[i]);
}
Stack<Integer> R = new Stack<>();
for (int i = n-1; i >= 0;i--){
ans[i] += R.size();//保存的是i往右看,能看到的全部楼高
if (i != n-1) {
while (!R.empty() && R.peek() <= arr[i]) {
R.pop();
}
}
R.push(arr[i]);
}
for (int i = 0; i < n; i++){
System.out.print((ans[i]+1)+" ");
}
}
}
类似问题:LeetCode每日温度
leetcode 739. Daily Temperatures
- 问题化简:找后面第一个比他大的元素。
- 首先想到从前往后遍历,这样会导致有大量重复计算的。
- 然后想到从后往前遍历,保存一个从上往下递增的栈,如果新来的元素比前面的元素比栈顶元素大的话,就会破坏栈的单调性,所以出栈,直到当前元素比栈顶元素大。
- 实现细节一:如果栈内保存的是温度,统计出栈次数作为间隔天数。这样这个栈只能统计出一次的结果。比如说 计算75 71 69 72。我们是从后往前遍历。求71的时候,栈里是69 72这时候把69出栈了,算出71的天数是2。那求75的时候怎么办?69已经没了啊!
- 所以栈里应该保存index,计算index之差作为间隔天数。
- 实现细节二:如果后面没有比他更高的温度,输出0。怎么判断呢? 好办,因为每次栈顶元素>=当前元素时(注意这个等号),出栈,如果栈都空了,自然就是没找到。
- 实现细节三:题干是找出更温暖的天,那温度<=当前温度的天都要被出栈,所以别忘了加等号。
package com.company;
import java.util.Stack;
class Solution {
public int[] dailyTemperatures(int[] T) {
int len = T.length;
int ans[] = new int[len];
Stack<Integer> stack = new Stack<>();
if (len <= 1){
return ans;
}
stack.push(len-1);
for (int i = len-2; i >= 0;i--){
while (!stack.empty() && T[i] > T[stack.peek()]){
stack.pop();
}
if (stack.empty()){
ans[i] = 0;
}else {
ans[i] = stack.peek()-i;
}
stack.push(i);
}
return ans;
}
}