【Lintcode】1066. Verify Preorder Serialization of a Binary Tree

题目地址：

https://www.lintcode.com/problem/1066/description

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

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

import java.util.Deque;
import java.util.LinkedList;

public class Solution {
    /**
     * @param preorder: a string
     * @return: return a bool
     */
    public boolean isValidSerialization(String preorder) {
        // write your code here
		String[] strs = preorder.split(",");
        
        Deque<String> stk = new LinkedList<>();
        for (int i = 0; i < strs.length; i++) {
            String s = strs[i];
            if (s.equals("#")) {
                while (!stk.isEmpty() && stk.peek().equals("#")) {
                	// pop出左边的null儿子
                    stk.pop();
                    // 如果栈空了，说明null儿子上面没有节点，这是不可能的，返回false
                    if (stk.isEmpty()) {
                        return false;
                    }
                    
                    // pop掉当前节点，下面的stk.push(s)会push一个null进去填补到左儿子那里
                    stk.pop();
                }
            }
            
            stk.push(s);
        }
        
        return stk.size() == 1 && stk.peek().equals("#");
    }
}

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