力扣—331验证二叉树的前序序列

题目描述

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

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

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

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

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

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

示例 1:

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

解题思路

根据官方题解的描述，可以使用槽位数slots来进行判断，首先使用split(',')方法将字符串变为列表进行存储，消耗O(N)的空间复杂度，对于上面的例子，从9开始遍历，9为根节点，首先初始化slots为1，表示初始只有一个槽位可用，遍历到9时，消耗掉一个槽位，所以slots -= 1，但是由于9不为“#”，所有要增加2个可用槽位，即slots += 2，最后判断可用槽位slots是否等于0即可。

class Solution(object):
    def isValidSerialization(self, preorder):
        nums = preorder.split(',')
        slots = 1
        for chr in nums:
            slots -= 1
            if slots < 0:
                return False
            if chr != '#':
                slots += 2
        return slots == 0

S = Solution()
print(S.isValidSerialization("9,3,4,#,#,1,#,#,2,#,6,#,#"))