~~~文末有原题出处,可以自己去做一把,体验一下~~~
给你一棵以 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 之间。
看到这个题目,直接拉笔就可以写出如下代码
type ListNode struct { Val int Next *ListNode}type TreeNode struct { Val int Left *TreeNode Right *TreeNode}func isSubPath(head *ListNode, root *TreeNode) bool { if head == nil { return true } if root == nil { return false } if root.Val == head.Val { return isSubPath(head.Next, root.Left) || isSubPath(head.Next, root.Right) || isSubPath(head, root.Left) || isSubPath(head, root.Right) } return isSubPath(head, root.Left) || isSubPath(head, root.Right)}
不过这个代码很不幸,超出了时间限制。
稍作优化之后,减少不必要的重复计算,得到一下代码。
type ListNode struct { Val int Next *ListNode}type TreeNode struct { Val int Left *TreeNode Right *TreeNode}func isSubPath(head *ListNode, root *TreeNode) bool { if root == nil { return false } return recur(head, root) || isSubPath(head, root.Left) || isSubPath(head, root.Right)}func recur(l *ListNode, node *TreeNode) bool { if l == nil { return true } if node == nil { return false } if l.Val != node.Val { return false } return recur(l.Next, node.Left) || recur(l.Next, node.Right)}
再次提交,执行通过。
每日一题,重新出发,看看这次能坚持多久。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/linked-list-in-binary-tree
留言与评论(共有 0 条评论) “” |