cubegao

Swift.栈的压入、弹出序列

2015-07-12

题目描述:输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

import Foundation

class For22Solution {
func isStackPushAndPop(_ pushStack: [Int],_ popStack: [Int]) -> Bool {

//辅助栈,取弹出序列的首位,等于栈顶就弹出。
var stack = [Int]()

var pushIndex = 0
var popIndex = 0

while popIndex <= popStack.count - 1 {

let popFirst = popStack[popIndex]
while stack.last != popFirst {

if pushIndex > pushStack.count - 1 {
return false
}

stack.append(pushStack[pushIndex])
pushIndex += 1
}

if stack.last == popFirst {
stack.popLast()
popIndex += 1
}
}

return true
}
}

算法思想:如果第二个序列中当前要判断的元素刚好与栈顶元素相等,则直接pop出来,如果不等,则将第一个序列的后面还没有入栈的元素入栈,直到将与之相等的元素入栈为止,如果第一个序列的所有的元素都入栈了,还没有找到与之相等的元素,则说明第二个序列不是第一个序列的弹出序列,

github地址:https://github.com/cubegao/LeetCode

Tags: 算法

扫描二维码,分享此文章