题目:
小Q在周末的时候和他的小伙伴来到大城市逛街,一条步行街上有很多高楼,共有n座高楼排成一行。
小Q从第一栋一直走到了最后一栋,小Q从来没有见过这么多的楼,所以他想知道他在每栋楼的位置处能看到多少栋楼呢?(当前面的楼的高度大于等于后面的楼时,后面的楼将被挡住)
输入描述:
第一行 楼栋数
第二行 每个楼的高度
输出:
输出一行,分别是小Q在第i栋楼上能看到楼的数量
方法:栈操作法
思路:
新建两个栈,为左栈和右栈
(1)当求所站这个楼左边能看到的楼栋数,就用栈存,每一个都需要判断,如果栈顶元素小于等于入栈的,就判断每个元素是否都小于等于入栈元素,小于就出去,大于就保留,然后插入入栈元素,如果栈顶元素大于入栈的,直接入栈
(2)求右边的楼栋,也用栈存,有点不一样,如果入栈的元素大于栈顶元素直接入栈,小于等于则不用入栈
(3)当在最左边和最右边只用一种,最后返回两个栈的大小之和+自己楼栋的1即可
代码:
public static int build(int[] array,int i){
Stack<Integer> left = new Stack<>();
Stack<Integer> right = new Stack<>();
if(i==0){
for (int j = 0; j <array.length ; j++) {
if(j==i) continue;
if(right.isEmpty()){
right.push(array[j]);
}else{
if(right.peek()<array[j]){
right.push(array[j]);
}
}
}
return right.size()+1;
}else if(i==array.length-1){
for (int j = 0; j <array.length ; j++) {
if(j==i) continue;
if(left.isEmpty()){
left.push(array[j]);
}else{
if(left.peek()<=array[j]){
while(!left.isEmpty()&&left.peek()<=array[j]){
left.pop();
}
left.push(array[j]);
}else{
left.push(array[j]);
}
}
}
return left.size()+1;
}else{
for (int j = 0; j <array.length ; j++) {
if(j==i){
continue;
}else if(j<i){
if(left.isEmpty()){
left.push(array[j]);
}else{
if(left.peek()<=array[j]){
while(!left.isEmpty()&&left.peek()<=array[j]){
left.pop();
}
left.push(array[j]);
}else{
left.push(array[j]);
}
}
}else{
if(right.isEmpty()){
right.push(array[j]);
}else{
if(right.peek()<array[j]){
right.push(array[j]);
}
}
}
}
return left.size()+right.size()+1;
}
}
public static void main(String[] args) {
int[] num = new int[]{5,3,8,3,2,5};
for (int i = 0; i <num.length ; i++) {
System.out.print(build(num,i));
}
}
代码比较复杂,有一些重复的,我这里分了三种情况讨论,其中的代码是重复的,比较杂乱