1367. 二叉树中的链表【二叉树】

1367. 二叉树中的链表

1367. 二叉树中的链表

给你一棵以 root 为根的二叉树和一个 head 为第一个节点的链表。

如果在二叉树中，存在一条一直向下的路径，且每个点的数值恰好一一对应以 head 为首的链表中每个节点的值，那么请你返回 True ，否则返回 False 。

一直向下的路径的意思是：从树中某个节点开始，一直连续向下的路径。

示例 1：

输入：head = [4,2,8], root = [1,4,4,null,2,2,null,1,null,6,8,null,null,null,null,1,3]
输出：true
解释：树中蓝色的节点构成了与链表对应的子路径。

示例 2：

输入：head = [1,4,2,6], root = [1,4,4,null,2,2,null,1,null,6,8,null,null,null,null,1,3]
输出：true

示例 3：

输入：head = [1,4,2,6,8], root = [1,4,4,null,2,2,null,1,null,6,8,null,null,null,null,1,3]
输出：false
解释：二叉树中不存在一一对应链表的路径。

提示：

• 二叉树和链表中的每个节点的值都满足 1 <= node.val <= 100 。
• 链表包含的节点数目在 1 到 100 之间。
• 二叉树包含的节点数目在 1 到 2500 之间。

解题思路

以每一个节点作为开头节点和链表比较，看看能否有路径可以和链表完全对上

Go代码

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func isSubPath(head *ListNode, root *TreeNode) bool {
    // 以每一个节点作为开头节点和链表比较，看看能否有路径可以和链表完全对上
    if head == nil {
        return true
    }
    if root == nil { // 经过上面的if,head不空，root却空了，说明当前不是有效路径
        return false
    }
    // 从当前节点开始与链表比较
    h := isValid(head,root)
    // 分别递归以左右节点开始与链表比较
    left := isSubPath(head,root.Left)
    right := isSubPath(head,root.Right)
    return h || left || right
    // 上面四行直接写成下面这一行即可，写成四行是为了更清楚的去思考递归
    // 实际上写成下面这一行会更好，因为是或，前面有一个是true，后面的递归都不用进行了，可以减少不少递归计算
    // return isValid(head,root) || isSubPath(head,root.Left) || isSubPath(head,root.Right)
}

func isValid(head *ListNode,cur *TreeNode) bool {
    // 链表遍历完了，说明是一条有效路径，可以不断往上返回true了
    if head == nil {
        return true
    }
    if cur == nil {
        return false
    }
    if cur.Val != head.Val {
        return false
    }
    // 当前节点对上了，继续用树的左右节点与链表的下一个节点比较
    return isValid(head.Next,cur.Left) || isValid(head.Next,cur.Right)
}