剑指 Offer 31. 栈的压入、弹出序列
题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
示例1:
输入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1] 输出:true 解释:我们可以按以下顺序执行: push(1), push(2), push(3), push(4), pop() -> 4, push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
示例2:
输入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2] 输出:false 解释:1 不能在 2 之前弹出。
数据限制:
0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed 是 popped 的排列
思路:
题目中给出了一个栈的入栈顺序和可能的出栈顺序,询问我们这个出栈顺序是否是正确的。
因为我们有栈的入栈顺序,我们可以直接根据出栈顺序进行模拟,如果温和,那么就说明顺序是正确的
我们可以用一个数组模拟栈内的空间,分别设置两个指针指向两个顺序
当我们入栈顺序的当前元素和出栈顺序的当前元素不同时,那么就将该元素入栈,并将指向入栈顺序的指针右移,如果两个元素相同,那么两个指针就都右移,并根据栈内空间的元素,和出栈顺序的首元素比较,如果相同,就将栈内空间的元素出栈,指向出栈顺序的指针右移,重复这个过程,知道栈内空间为空,或元素不同。最后,当元素全部入栈之后,将栈内空间的所有元素出栈,判断是否和出栈顺序相同, 如果相同,则出栈顺序是正确的,否则是错误的
时间复杂度:O(n)
空间复杂度:O(n)
代码:
class Solution { public boolean validateStackSequences(int[] pushed, int[] popped) { if (pushed.length != popped.length) return false; int n = pushed.length, l = 0, r = 0, p = 0; if (n == 0) return true; int[] sta = new int[n]; sta[p] = pushed[l]; p++;l++; while(l < n){ while (p > 0 && sta[p - 1] == popped[r]){ p--; r++; } sta[p] = pushed[l]; p++;l++; } while (p > 0 && sta[p - 1] == popped[r]){ p--; r++; } if (r == n) return true; else return false; } }