【Leetcode】331. Verify Preorder Serialization of a Binary Tree

题目地址：

https://leetcode.com/problems/verify-preorder-serialization-of-a-binary-tree/

给定一棵二叉树的前序遍历，问其是否合法。该前序遍历是含null节点的，null以'#'表示。其余节点以数字表示，以','隔开。

法1：递归。直接模拟以二叉树前序遍历建树的过程就行了。先以","做split，这样就得到了每个节点元素。接着开始模拟建树，并用一个下标来跟踪用到了哪个节点。如果该建立节点但是无元素可用了，则返回false；如果是null，则建出null节点，后移指针并返回true，否则说明是一个非null节点，建立之指针后移，然后递归建立左子树和右子树，如果其一建立失败就返回false，否则返回true。代码如下：

class Solution {
public:
  bool isValidSerialization(string t) {
    istringstream ss(t);
    vector<string> v;
    for (string s; getline(ss, s, ',');)
      v.push_back(s);
    int idx = 0;
    auto dfs = [&](this auto &&dfs) -> bool {
      if (idx == v.size())
        return false;
      if (v[idx] == "#") {
        idx++;
        return true;
      }

      idx++;
      return dfs() && dfs();
    };
    return dfs() && idx == v.size();
  }
};

时间复杂度$O(n)$，$n$是字符串长度，空间$O(h)$，$h$是树高。

法2：栈。可以用栈模拟前序遍历，主要问题是遍历到null的时候怎么判断它是左儿子还是右儿子。这里采用这种办法，每次遍历到null节点的时候，看一下栈顶，如果栈顶不是null，说明此null是左儿子，则入栈；如果栈顶是null，那么说明当前的null是右儿子，那么就pop掉栈顶的null，接下来栈顶应该是元素值（如果此时栈空了说明不合法），将其pop掉，接下来如果栈顶还是null，则说明刚才pop掉一个右儿子，接下来需要继续pop出一个元素，则pop掉null再pop掉元素值，如此循环直到栈顶不是null，接下来再把遍历到的null入栈，这个null是标记左儿子用的，表示栈顶的左子树已经遍历完毕。这个while循环是在将一条右链全pop出来。代码如下：

class Solution {
public:
  bool isValidSerialization(string t) {
    istringstream ss(t);
    vector<string> v;
    for (string s; getline(ss, s, ',');)
      v.push_back(s);
    stack<string> stk;
    for (string &s : v) {
      if (s == "#") {
        while (stk.size() && stk.top() == "#") {
          stk.pop();
          if (stk.empty()) return false;
          stk.pop();
        }
      }
      stk.push(s);
    }

    return stk.size() == 1 && stk.top() == "#";
  }
};

时空复杂度一样。