跳至主要內容

31. 栈的压入、弹出序列


31. 栈的压入、弹出序列open in new window

🟠   🔖  数组 模拟  🔗 力扣open in new window

题目

现在图书馆有一堆图书需要放入书架,并且图书馆的书架是一种特殊的数据结构,只能按照 一定 的顺序 放入拿取 书籍。

给定一个表示图书放入顺序的整数序列 putIn,请判断序列 takeOut 是否为按照正确的顺序拿取书籍的操作序列。你可以假设放入书架的所有书籍编号都不相同。

示例 1:

输入:putIn = [6,7,8,9,10,11], takeOut = [9,11,10,8,7,6]

输出:true

解释:我们可以按以下操作放入并拿取书籍:

push(6), push(7), push(8), push(9), pop() -> 9,

push(10), push(11),pop() -> 11,pop() -> 10, pop() -> 8, pop() -> 7, pop() -> 6

示例 2:

输入:putIn = [6,7,8,9,10,11], takeOut = [11,9,8,10,6,7]

输出:false

解释:6 不能在 7 之前取出。

提示:

  • 0 <= putIn.length == takeOut.length <= 1000
  • 0 <= putIn[i], takeOut < 1000
  • putIntakeOut 的排列。

注意

本题与 LeetCode 第 946 题 相同。

解题思路

这道题可以使用模拟栈的方法进行判断,使用一个辅助栈 stack,模拟 pushpop 操作,最后通过判断栈是否为空来得到最终的结果。

具体思路如下:

  1. 遍历 putIn 数组,模拟入栈操作,并在每次入栈后,判断是否需要执行出栈操作。
  2. 如果当前栈顶元素等于 takeOut 数组中下一个要出栈的元素,则执行出栈操作。
  3. 遍历结束后,判断栈是否为空。

代码

/**
 * @param {number[]} putIn
 * @param {number[]} takeOut
 * @return {boolean}
 */
var validateBookSequences = function (putIn, takeOut) {
	let stack = [];
	let popIndex = 0;

	for (let item of putIn) {
		stack.push(item);
		while (stack.length > 0 && stack[stack.length - 1] == takeOut[popIndex]) {
			stack.pop();
			popIndex++;
		}
	}
	return stack.length == 0;
};