算法: 在数组中找到左边数都比它小, 右边数都比它大的数

今天是转正二面, Leader跟我聊了2个小时, 虽然都是轻松地聊, 但我还是好紧张呀🙈主要还是太担心Leader对我的看法了...希望以后能够改正. 下面记录一下今天Leader考我的这道算法题

题目

在无序不重复数组中寻找满足如下条件的值:

• 该值左边的数均不大于它
• 该值右边的数均不小于它

解法

思路是使用栈来存满足前面的值均小于当前值的值. 因需要前面的值都小于该值, 需要用一个max来存到目前为止的最大值, 若当前值大于max, 则满足第一个条件. 若后续出现更小的值, 则之前压入栈中的值一定不满足第二个条件, 因此应出栈.

总结一下就是: 对于数组中的每个数num, 先检查是否更小, 若当前值比栈中元素更小, 则之前压入栈中的值不再满足条件, 需出栈; 若当前值比目前的max更大, 则满足条件1, 可以先入栈.

java代码如下:

    private static int[] midValue(int[] nums){
        if(0 == nums.length) {return null;}
        // 存储当前满足条件的值
        Stack<Integer> midNumsStack = new Stack<>();
        // 记录当前遍历过的最大值(新入栈的数必须不小于max)
        int max = nums[0];
        midNumsStack.push(nums[0]);
        for(int i=1; i<nums.length; i++){
            // 先检查新值num是否小于栈内元素, 否则栈内元素不满足条件
            while(!midNumsStack.empty() && nums[i] <= midNumsStack.peek()){
                midNumsStack.pop();
            }
            if(nums[i] > max){
                midNumsStack.push(nums[i]);
                max = nums[i];
            }
        }
        int[] result = new int[midNumsStack.size()];
        int i = 0;
        while(midNumsStack.size()>0){
            result[i++] = midNumsStack.pop();
        }
        return result;
    }
复制代码

回来后, 一起实习的清华同学问我面试详情, 我把这道题告诉他了, 他立刻就有了思路, 说是遍历两遍. 虽然我现在也明白了这种思路(一遍找符合前面都小于的数, 后一遍从后向前找符合后面都大于的数, 两次均存在的数即为目标值). 我感觉自己好笨, Leader问后我想了好久, 回来后直接看了答案. 以后还要好好刷题呀!