剑指offer 31. 栈的压入、弹出序列/leetcode 946. 验证栈序列 golang实现

描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如，序列 {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 之前弹出。

思路

1. pushIndex popIndex 分别表示pushed pupped的索引  stack表示我们的栈
2. 判断当前栈顶是否等于popped[popIndex]
    1) 等于则出栈且popIndex++
    2) 不等于则将pushed[pushIndex]入栈 且pushIndex++ 
3. 循环2步骤直到所有数都入栈 即pushIndex==len(pushed) 
4. 继续循环1 直到所有数都出栈 即popIndex==len(popped)
5. 判断是否所有数都入栈 所有数都出栈 即 pushIndex == popIndex

实现

func validateStackSequences(pushed []int, popped []int) bool {
     
    sizePush := len(pushed)
    sizePop := len(popped)
    
    if sizePush != sizePop{
     
        return false
    }else if sizePop == sizePush && sizePop == 0{
     
        return true
    }
    var stack = make([]int, 0)
    var pushIndex int
    var popIndex int
    

    for pushIndex < sizePush || popIndex < sizePush{
     
        if len(stack) > 0 && stack[len(stack) - 1] == popped[popIndex]{
     
            stack = stack[0:len(stack)-1]
            popIndex++
        }else if pushIndex < sizePush{
     
            stack = append(stack, pushed[pushIndex])
            pushIndex++
        }else{
     
            break
        }
    }


    if pushIndex != popIndex{
     
        return false
    }

    return true
}