问题描述
给你一棵以 root 为根的二叉树和一个 head 为第一个节点的链表。
如果在二叉树中,存在一条一直向下的路径,且每个点的数值恰好一一对应以 head 为首的链表中每个节点的值,那么请你返回 True ,否则返回 False 。
一直向下的路径的意思是:从树中某个节点开始,一直连续向下的路径。
Example
![]()
输入:head = [4,2,8], root = [1,4,4,null,2,2,null,1,null,6,8,null,null,null,null,1,3]
输出:true
解释:树中蓝色的节点构成了与链表对应的子路径。
Note
- 二叉树和链表中的每个节点的值都满足 1 <= node.val <= 100 。
- 链表包含的节点数目在 1 到 100 之间。
- 二叉树包含的节点数目在 1 到 2500 之间。
题目链接:1367. 二叉树中的列表 (难度:中等)
思路
为了完成二叉树和链表的匹配任务,我们需要对 root 中的每一棵子树进行匹配。因此,我们需要实现两个函数,一个用于匹配当前以 root 为根的子树 (helper),另一个函数用于枚举整棵树中的所有子树(isSubPath)。由于路径本身指定了从上到下,这两个函数均采用先序遍历的方式进行处理
代码
class Solution {
public:
bool helper(ListNode* head, TreeNode* root){
if(head == NULL)
return true;
if(root == NULL)
return false;
if(root->val == head->val){
return helper(head->next, root->left) || helper(head->next, root->right);
}
return false;
}
bool isSubPath(ListNode* head, TreeNode* root) {
if(head == NULL)
return true;
if(root == NULL)
return false;
if(helper(head, root))
return true;
return isSubPath(head, root->left) || isSubPath(head, root->right);
}
};
执行结果: 52 ms, 22.1 MB
网友评论