[路飞]_leetcode_331.验证二叉树的前序序列化

107 阅读2分钟

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

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

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

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

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

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

解题思路

槽位方式

  1. 当遇到非空节点,占用一个槽位,添加2个空槽位
  2. 当遇到空节点,占用一个槽位。
  3. 当槽位占满时,删除当前位置。

出度入度方式

  1. 根节点的出度为2,入度为0
  2. 非空非根节点 出度为2, 入度为1
  3. 空节点的出度为0 入度为1

在正常情况下,出度 == 入度。如果不想等表示二叉树不正确

代码

槽位方式

var isValidSerialization = function(preorder) {
    let stack = [1];
    let arr = preorder.split(',');
    for (let i = 0; i < arr.length; i++) {
        // 如果中间有槽位不够情况,则表示二叉树不正确
        if (!stack.length) return false
        // 不管是否为空节点都会先占用一个槽位
        stack[stack.length-1]--
        // 如果槽位占满,则移除当前位置
        if (stack[stack.length - 1] == 0) stack.pop()
        if (arr[i] !== '#') {
            // 非空节点添加2个槽位
            stack.push(2)
        }
    }

    return !stack.length
};

出度入度

var isValidSerialization = function(preorder) {
    let arr = preorder.split(',')
    let diff = 1; // 根节点没有入度
    for (let i = 0; i < arr.length; i++) {
        // 节点入度为1
        diff--
        if (diff < 0) return false
        if (arr[i] !== '#') {
            // 非空节点出度为2
            diff += 2
        }
    }
    return diff == 0;
};