Swift.栈的压入、弹出序列

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

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

标签: 算法

非特殊说明,本博所有文章均为博主原创。

评论啦~