笔试之栈和队列总结

今天来总结一下栈和队列，不多说直接看题

• 最小栈：（https://leetcode-cn.com/problems/min-stack/）

       

 

 要求求最小元素要在O(1)时间复杂度内，但是一般求最小元素需要遍历栈至少O(n)，这里就会想到用空间换时间，使用两个栈来实现最小栈，看代码：

public class _155最小栈 {
    private Stack<Integer> stack;
    private Stack<Integer> stack1;
    public _155最小栈() {
       stack = new Stack<>();
       stack1 = new Stack<>();
    }
    public void push(int x) {
        stack.push(x);
        if(stack1.isEmpty()){
            stack1.push(x);
        }else {
                stack1.push(Math.min(x,stack1.peek()));
        }
    }
    public void pop() {
        stack.pop();
        stack1.pop();
    }
    public int top() {
        return  stack.peek();
    }
    public int getMin() {
        return stack1.peek();
    }
}

• 滑动窗口最大值（https://leetcode-cn.com/problems/sliding-window-maximum/）

 

这个题首先可以用暴力，但是会超时，也可以用队列，但是这里介绍一种更巧妙地方法，看代码，其实主要还是维护最大值的索引

    public static  int[] maxSlidingWindow(int[] nums, int k) {
        int[] dp = new int[nums.length - k + 1];
        int maxIndex = 0;
        for (int i = 1; i < k; i++) {
            if(nums[i] > nums[maxIndex]) maxIndex = i;
        }
        for (int i = 0; i < dp.length; i++) {
            int r = i + k -1;
            if(maxIndex < i){//最大值索引不在窗口内，进行重新找最大索引
                maxIndex = i;
                for (int j = i + 1; j <= r; j++) {
                    if(nums[j] > nums[maxIndex]) maxIndex = j;
                }
            }else  if (nums[r] > nums[maxIndex]){
                maxIndex = r;
            }
            dp[i] = nums[maxIndex];
        }
        return dp;
    }