刷题---树篇---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 之间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/linked-list-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

______________________________________________________________________________________________________

沾点回溯的思想

在二叉树里找一条连续的路径，路径上的值与链表的值相同。

那么可以分三种情形：

（1）当未匹配到链表的头节点时，这种情形下链表不同，二叉树向下遍历；

（2）当匹配到链表的头节点时，这种情形下链表和二叉树都同时向下遍历；

（3）这种情况是基于情况（2）的条件下，当二叉树与链表的某一节点完全不匹配时，此时继续遍历二叉树，但链表回到头节点，重复情况（1）。

所以可以用一个变量head记录链表的头节点，curr记录链表现在的节点则,用flag标记并记录结果:

    //寻找链表所在的节点与二叉树所在的节点是否匹配
    //若匹配成功则链表和二叉树都向下遍历
    //情况(2)
    if curr.Val == root.Val {
        Run(head,curr.Next,root.Left,flag)
        Run(head,curr.Next,root.Right,flag)
    } 
    //判断链表是否从头开始匹配
    //若还未匹配到链表的头节点，则只遍历二叉树
    //情况(1)
    if head == curr {
        Run(head,head,root.Left,flag)
        Run(head,head,root.Right,flag)
    }

结束条件：

    if curr == nil  {
        *flag = true
        return
    }
    if root == nil || *flag == true {
        return 
    }

完整代码:

func isSubPath(head *ListNode, root *TreeNode) bool {
    flag := false

    Run(head,head,root,&flag)

    return flag
}

func Run(head,curr *ListNode, root *TreeNode , flag *bool) {
    if curr == nil  {
        *flag = true
        return
    }
    if root == nil || *flag == true {
        return 
    }
    
    if curr.Val == root.Val {
        Run(head,curr.Next,root.Left,flag)
        Run(head,curr.Next,root.Right,flag)
    }
    //如果程序从if curr.Val == root.Val{}中出来，再往下执行，实际上是情形(3)
    //相当于从第一次匹配到链表头节点的位置按情形(1)往下执行
    
    if head == curr {
        Run(head,curr,root.Left,flag)
        Run(head,curr,root.Right,flag)
    }
}