331. 验证二叉树的前序序列化

 

331. 验证二叉树的前序序列化

链接：https://leetcode-cn.com/problems/verify-preorder-serialization-of-a-binary-tree/

序列化二叉树的一种方法是使用前序遍历。当我们遇到一个非空节点时，我们可以记录下这个节点的值。如果它是一个空节点，我们可以使用一个标记值记录，例如 #。

     _9_
    /   \
   3     2
  / \   / \
 4   1  #  6
/ \ / \   / \
# # # #   # #

例如，上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#"，其中 # 代表一个空节点。

给定一串以逗号分隔的序列，验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。

每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 '#' 。

你可以认为输入格式总是有效的，例如它永远不会包含两个连续的逗号，比如 "1,,3" 。

示例 1:

输入: "9,3,4,#,#,1,#,#,2,#,6,#,#"
输出: true

示例 2:

输入: "1,#"
输出: false

示例 3:

输入: "9,#,#,1"
输出: false

核心思路：我们知道，对于一个非空节点而言，它提供了左右孩子两个可以放置其他节点或空节点的位置，因此我们只要计算到最后，所有的位置都被空节点或节点占满即可。

思路1：题目不让我们重建树结构，我们可以用栈进行模拟，每读入一个新节点就要将上一个节点的剩余位置数减1，若减法后为0，表示上一个节点的左右孩子都被安排满，则可弹出；同时，如果新节点为#，则不提供有效位，如果新节点为数字，则提供两个有效位，注意初始化的时候，根节点只有一个，因此空树时只提供一个有效位。

class Solution {
public:

    bool isValidSerialization(string preorder) {
        int n = preorder.length(), i = 0;
        stack<int> ST;
        ST.push(1);
        while(i < n){
            if(ST.empty())return false;
            if(preorder[i] == '#'){
                ST.top() -= 1;
                if(ST.top() == 0){
                    ST.pop();
                }
                i += 2; // 跳过逗号
            }else {
                while(i < n && isdigit(preorder[i])){
                    ++ i;
                }
                ST.top() -= 1;
                if(ST.top() == 0){
                    ST.pop();
                }
                ST.push(2); // 一个节点占用一个位置 并提供两个位置
                ++ i; // 跳过逗号
            }
        }
        return ST.empty();
    }

};

思路2：我们直接用变量存储下剩余的位置数。

class Solution {
public:
    bool isValidSerialization(string preorder) {
        int n = preorder.length(), i = 0, rest = 1;
        while(i < n){
            if(!rest) return false; // 执行过程中空位置不足
            if(preorder[i] == '#'){
                -- rest;
                i += 2; // 跳过逗号
            }else {
                while(i < n && isdigit(preorder[i])){
                    ++ i;
                }
                ++ rest; // 一个节点占用一个位置 并提供两个位置
                ++ i; // 跳过逗号
            }
        }
        return rest == 0;
    }
};