题目链接
牛客网在线oj题——栈的压入,弹出序列
题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
- 0<=pushV.length == popV.length <=1000
- -1000<=pushV[i]<=1000
- pushV 的所有数字均不相同
题目示例
示例1
输入:
[1,2,3,4,5],[4,5,3,2,1]
返回值:
true
说明:
可以通过push(1)=>push(2)=>push(3)=>push(4)=>pop()=>push(5)=>pop()=>pop()=>pop()=>pop()
这样的顺序得到[4,5,3,2,1]这个序列,返回true
示例2
输入:
[1,2,3,4,5],[4,3,5,1,2]
返回值:
false
说明:
由于是[1,2,3,4,5]的压入顺序,[4,3,5,1,2]的弹出顺序,要求4,3,5必须在1,2前压入,且1,2不能弹出,但是这样压入的顺序,1又不能在2之前弹出,所以无法形成的,返回false
解题思路
首先,如果pushA或者popA为空,或者两者的长度不同,肯定就不可能是该栈的弹出序列,直接返回false
可以直接创建一个栈,定义一个i元素用来遍历pushA序列,定义一个j元素用来遍历popA序列
先将pushA[i]元素插入栈,如果栈不为空并且栈顶元素值等于popA[j],那么就一直弹出栈顶元素,j++
重复上述过程,如果最后栈不为空,说明还有元素未被弹出,也就返回false
例如:
首先插入push[i]元素入栈
此时pop[j] != stack.peek(),I++,继续将arr[i]压入栈
此时pop[j] != stack.peek(),I++,继续将arr[i]压入栈
当i为3时,此时压入栈元素为4,pop[j] == stack.peek(),弹出栈顶元素,j++
此时pop[j] != stack.peek(),I++,继续压入栈顶
此时pop[j] == stack.peek(),弹出栈顶元素,j++,一直重复上述操作,直到栈为空
最后当i遍历完整个pushA数组时,stack为空,说明popA是正确的弹出序列,返回true
完整代码
import java.util.*;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
if(pushA == null || popA == null || pushA.length != popA.length){
return false;
}
Stack<Integer> stack = new Stack<>();
int j = 0;
for (int i = 0; i < pushA.length; i++){
stack.push(pushA[i]);
while(!stack.isEmpty() && stack.peek() == popA[j]){
stack.pop();
j++;
}
}
return stack.isEmpty();
}
}