【路飞】算法与数据结构-验证二叉树的前序序列化

138 阅读1分钟

LeetCode:地址

题目要求

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

     _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
思路

一个非空节点有两个子节点, 空节点没有子节点

代码
/**
 * @param {string} preorder
 * @return {boolean}
 */
var isValidSerialization = function(preorder) {
    let arr = preorder.split(',')
    let stack = [1]
    for(let i = 0; i < arr.length; i++){
        if (!stack.length) return false
        stack.pop()
        if (arr[i] !== '#') stack.push(1, 1)
    }
    return stack.length === 0
};

降维


var isValidSerialization = function(preorder) {
    let arr = preorder.split(',')
    let num = 1
    for(let i = 0; i < arr.length; i++){
        if ( num <= 0 ) return false
        num--
        if (arr[i] !== '#') num += 2
    }
    return num === 0
};