单调栈

单调栈,如名字一样,栈内的元素是单调递增或者单调递减的。

接下来我们用 LeetCode 的题目 155. Min Stack 来说明这种特殊的数据结构,题目说明如下:

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

  • push(x) -- Push element x onto stack.
  • pop() -- Removes the element on top of the stack.
  • top() -- Get the top element.
  • getMin() -- Retrieve the minimum element in the stack.

Constraints: Methods pop, top and getMin operations will always be called on non-empty stacks.

显然,在这到题目中,我们需要维持一个单调递减的栈,这样我们才能在 constant time 内获取到最小元素。

这道题目的思路,就是我们需维护两个栈,栈 stack 保存原来的元素,栈 minStack 保存入栈元素的单调性,在我们新入栈元素的时候,只要新元素比栈 minStack 内的元素小我们就可以入栈。

这里需要特别注意,我们不要把栈 minStack 内元素出栈。因为栈的特性就是在一端操作。

/**
 * @param {number} x
 * @return {void}
 */
MinStack.prototype.push = function (x) {
  this.stack.push(x);

  // 这一步操作是有问题的,栈minStack与栈stack的元素序号乱了
  // while (this.minStack.length && this.minStack[this.minStack.length - 1] > x) {
  //  this.minStack.pop();
  // }
  // this.minStack.push(x);

  // 新入栈元素小的,即可入栈
  if (!this.minStack.length || x <= this.minStack[this.minStack.length - 1]) {
    this.minStack.push(x);
  }
};

在我们出栈时候,需要同步把 minStack 内元素也做出栈检查。

/**
 * @return {void}
 */
MinStack.prototype.pop = function () {
  let top = this.stack.pop();
  if (top === this.minStack[this.minStack.length - 1]) this.minStack.pop();
};

单调栈的完整代码如下:

/**
 * initialize your data structure here.
 */
const MinStack = function () {
  this.stack = [];
  this.minStack = [];
};

/**
 * @param {number} x
 * @return {void}
 */
MinStack.prototype.push = function (x) {
  this.stack.push(x);
  // while (this.minStack.length && this.minStack[this.minStack.length - 1] > x) {
  //  this.minStack.pop();
  // }
  // this.minStack.push(x);
  if (!this.minStack.length || x <= this.minStack[this.minStack.length - 1]) {
    this.minStack.push(x);
  }
};

/**
 * @return {void}
 */
MinStack.prototype.pop = function () {
  let top = this.stack.pop();
  if (top === this.minStack[this.minStack.length - 1]) this.minStack.pop();
};

/**
 * @return {number}
 */
MinStack.prototype.top = function () {
  return this.stack[this.stack.length - 1];
};

/**
 * @return {number}
 */
MinStack.prototype.getMin = function () {
  if (this.minStack.length) {
    return this.minStack[this.minStack.length - 1];
  }
  return this.stack[this.stack.length - 1];
};
posted @ 2020-11-16 11:03  浪荡&不羁  阅读(77)  评论(0编辑  收藏  举报