搞定八道高频算法题

一、如何找右边第一个比我小的元素

二、如何找右边第一个比我大的元素

三、如何找右边最后一个比我小的元素

四、如何找右边最后一个比我大的元素

五、如何找左边第一个比我小的元素

六、如何找左边第一个比我大的元素

七、如何找左边最后一个比我小的元素

八、如何找左边最后一个比我大的元素

首先，我们先记住每个元素右边第一个比我小的元素索引，如果没有标记为-1如下图所示

接着，我们扩展出下图所示

 

首先，当我们遍历第一个元素1的时候，我们不知道后边第一个比它小的元素，所以先记住索引0，等后面遇到比它小的元素再更新索引， 接着遍历第二个元素2的时候，因为它比第一个元素1大，所以依旧记住索引1， 接着遍历第三个元素3的时候，因为它比第二个元素2大，所以依旧记住索引2， 接着遍历第四个元素9的时候，因为它比第三个元素3大，所以依旧记住索引3，接着到遍历第五个元素5的时候，发生了变化，因为此时它比第四个元素9小，所以需要更新索引把3更新为4，并且对应的res[3]=4，但是它比第三个元素3大，所以记住索引4，接着到遍历第六个元素0的时候，发生了变化，因为此时它比第五个元素5小，所以需要更新索引把4更新为5，并且对应的res[4]=5，继续它比第三个元素3小，所以需要更新索引把2更新为5，res[2]=5，继续它比第二个元素2小，所以需要更新索引把1更新为5，res[1]=5，

继续它比第一个元素1小，所以需要更新索引把0更新为5，res[0]=5。此时这个过程类似于单调栈，因为栈为空了，那么第六个元素即变成第一个元素，我们不知道后边第一个比它小的元素，所以先记住索引5，等后面遇到比它小的元素再更新索引，接着到遍历第七个元素6的时候，因为它比第六个元素0大，所以依旧记住索引6，遍历结束。因为找不到右边第一个比它小的元素，那么对应的res[5]=-1,res[6]=-1

public int findRightFirstSmall(int[] nums){
    int[] res= new int[nums.length];
    ArrayDeque<Integer> stack=new ArrayDeque<>();
    for(int i=0;i<nums.length;i++){
        int x=nums[i];
        // 找出右边第一个比它小的元素并更新索引
        while(!stack.isEmpty()&&x<nums[stack.peek()]){
            res[stack.peek()]=i;
            stack.pop();
        }
        stack.push(i);
    }
    while(!stack.isEmpty()){
        res[stack.peek()]=-1;
        stack.pop();
    }
    return res;
}