leetcode_力扣_剑指 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 之前弹出。

提示

1. 0 <= pushed.length == popped.length <= 1000
2. 0 <= pushed[i], popped[i] < 1000
3. pushed 是 popped 的排列。

解题思路

1. 将 pushed 中的输入重新压栈一遍，同时也与 popped 中的第一个元素进行比较；
2. 相等则出栈，将 popped 中第一个元素删掉，重读第 1 步；
3. 不相等则继续压栈，重复第 1 步，直至将所有元素全部压栈；
4. 待到所有元素都压栈后，如果 popped 中还有元素，则说明顺序有问题，否则说明合理。

代码

class Solution {
public:
    bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {

        int popped_len = popped.size();		// 获取元素个数，两个 vector 内元素个数是一样的
        if(popped_len == 0){	// 如果元素个数为 0，则直接返回 true
            return true;
        }
        int pop_idx = 0;	// 指向 popped 中将要 pop 的元素
        int i = 0;

        while(i < popped_len){		// 这里模拟了一个栈，i 为栈顶指针
        
            while(pop_idx < popped_len && popped[pop_idx] == pushed[i]){	// 用栈顶元素与 popped 中将要 pop 的元素比较
            	// 如果两个元素相同，则模拟 pop 后的操作
                pop_idx++;
                if(pop_idx == popped_len){	// 当 popped 中所有元素都出栈，直接返回 true
                    return true;
                }
                pushed[i] = -1;		// 将 pop 的元素在 pushed 里置 -1，表示此处为空
                do{
                    i--;
                }while(i > -1 && pushed[i] == -1);	// 确定下一个栈顶指针 i 的位置

                while(i < 0 || pushed[i] == -1){	// i 可能为负数，是的话再次确认一遍位置
                    i++;
                }
            }

            i++;	// 模拟一次压栈

        }

        return false;	// 程序能走到这块，就说明出栈的顺序存在问题
    }
};

提交结果

总结

• 操作栈时，注意判断栈空栈满和栈顶指针的位置，其他操作都是中规中矩的。