「leetcode系列」第三十二期 二叉树中的列表

~~~文末有原题出处,可以自己去做一把,体验一下~~~

给你一棵以 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 条评论) “”
   
验证码:

相关文章

推荐文章