leetcode_1430 判断给定的序列是否是二叉树从根到叶的路径

128 阅读1分钟

要求

给定一个二叉树,我们称从根节点到任意叶节点的任意路径中的节点值所构成的序列为该二叉树的一个 “有效序列” 。检查一个给定的序列是否是给定二叉树的一个 “有效序列” 。

我们以整数数组 arr 的形式给出这个序列。从根节点到任意叶节点的任意路径中的节点值所构成的序列都是这个二叉树的 “有效序列” 。

示例 1:

image.png

输入:root = [0,1,0,0,1,0,null,null,1,0,0], arr = [0,1,0,1]
输出:true
解释:
路径 0 -> 1 -> 0 -> 1 是一个“有效序列”(图中的绿色节点)。
其他的“有效序列”是:
0 -> 1 -> 1 -> 0 
0 -> 0 -> 0

示例 2:

image.png

输入:root = [0,1,0,0,1,0,null,null,1,0,0], arr = [0,0,1]
输出:false 
解释:路径 0 -> 0 -> 1 不存在,所以这不是一个“序列”。

示例 3:

image.png

输入:root = [0,1,0,0,1,0,null,null,1,0,0], arr = [0,1,1]
输出:false
解释:路径 0 -> 1 -> 1 是一个序列,但不是一个“有效序列”(译者注:因为序列的终点不是叶节点)。

提示:

  • 1 <= arr.length <= 5000
  • 0 <= arr[i] <= 9
  • 每个节点的值的取值范围是 [0 - 9]

核心代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.arr = []
    
    def dfs(self,node,i):
        if node and i < len(self.arr) and node.val == self.arr[i]:
            if i == len(self.arr) - 1 and not node.left and not node.right:
                return True
            else:
                return self.dfs(node.left,i + 1) or self.dfs(node.right,i + 1)
        else:
            return False

    def isValidSequence(self, root: TreeNode, arr: List[int]) -> bool:
        self.arr = arr
        return self.dfs(root,0)

image.png

解题思路:这个题就是普通的二叉树的深度遍历,值得学习的地方就是我么在进行深度遍历的时候增加了一些附加条件,可以快速的过滤掉不是路径的深度路径检查,减少时间的耗费。