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

问题描述

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

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

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

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

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

问题链接地址

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

示例

示例1

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

示例2

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

示例3

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

问题标签

栈，树

解题思路—栈

根据‘#’遍历，遇到‘#’，层数减一，数字加一，返回最后结果是否遍历完全，即最后的层数是否等于0.

代码

class Solution:
    def isValidSerialization(self, preorder: str) -> bool:
        n = 1 #初始化树的层数
        preorder = preorder.split(',')
        for i in preorder:
            if n == 0:return False
            if i == '#':#遍历到空节点，返回到上一层
                n -= 1
                if n < 0:return False #如果层数小于0，返回False
            else:
                n += 1 #非空节点，继续向下遍历，层数+1
        return n == 0