【Leetcode】946. Validate Stack Sequences

题目地址：

https://leetcode.com/problems/validate-stack-sequences/

给定一个入栈序列，再给定另一个序列，问其是否可能是一个合法的出栈序列。题目保证入栈序列数字彼此不同。

思路是直接用一个栈去模拟。先将入栈序列里的数入栈，然后中途如果发现栈顶等于了出栈序列里的值，则不停出栈。看最后操作完之后栈是否为空，若非空则说明序列不是合法的出栈序列。代码如下：

class Solution {
 public:
  bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
    stack<int> stk;
    int idx = 0;
    for (int x : pushed) {
      stk.push(x);
      while (stk.size() && stk.top() == popped[idx]) {
        stk.pop();
        idx++;
      }
    }

    return stk.empty();
  }
};

时空复杂度$O(n)$。

算法正确性证明：
容易知道如果最后栈空，则序列一定是合法的出栈序列；现在尝试证明另一个方向。首先将pushed序列里的数字按顺序与$1,2,...,n$做一个双射$\sigma$，用这个$\sigma$对popped序列也做一个双射，容易知道做了双射后问题是等价的。所以问题就变为，已知入栈序列是$(1,2,3,...,n)$，则上述算法对于合法出栈序列而言，最后栈一定是空的。这里需要用到一个结论：一个序列是合法出栈序列当且仅当其没有“$312$”模式。“$312$”模式是指，存在三个数$x_i,x_j,x_k$使得$i<j<k$，但是$x_i>x_k>x_j$。具体证明，可以参考https://blog.csdn.net/qq_46105170/article/details/105837196代码后附的证明。有了这个结论，我们来证明那个命题。数学归纳法。如果给定的合法出栈序列长度为$1$、$2$或者$3$，已知没有“$312$”模式，接下来可以用暴力枚举知道算法最后栈是空的。假设这一结论对于序列长度为$n$时也是对的，当长度为$n+1$的时候，我们考虑popped的最后一个数，设其为$y$，然后再在pushed里找到$y$。由于$y$是最后一个出栈的数，所以在pushed序列里，$y$之前的数是已经pop干净了以后才把$y$进栈的。所以把pushed序列分为两部分，前半部分是$1,2,...,y-1$，后半部分是$y,y+1,...,n+1$，同样也可以把popped序列分为连续的两部分（这里必然是连续的两部分，否则个数就对不上了），前半部分小于$y$，后半部分大于等于$y$。由归纳假设，小于$y$的部分由于是合法出栈序列（之所以是合法出栈序列，是因为全局上没有"$312$"模式，所以前半部分当然也没有"$312$"模式），所以遍历完后，算法里的栈会空，后半部分也是合法出栈序列，所以遍历完后，栈也会空，也就是最后栈会空。由数学归纳法知道，命题对任意长度的合法出栈序列都对。