剑指offer--树的子结构

最新推荐文章于 2021-11-18 19:13:14 发布

原创最新推荐文章于 2021-11-18 19:13:14 发布 · 83 阅读

0 ·

CC 4.0 BY-SA版权

c++ 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否是另一棵二叉树A的子结构。通过遍历树A并寻找与树B根节点值相同的节点，然后递归比较两个树的结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
/*
第一在树A中遍历找到和树B的根节点的值一样的节点R；
第二判断树A中以R为根节点的子树是不是包含树B一样的结构。
*/
class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(!pRoot1 || !pRoot2)//当Tree1和Tree2都不为零的时候，才进行比较。否则直接返回false
            return false;
        bool result=false;
        if(pRoot1->val == pRoot2->val)//如果找到了对应Tree2的根节点的点
            result=isSubtree(pRoot1,pRoot2); //以这个根节点为为起点判断是否包含Tree2
        if(!result)    //如果找不到，那么就再去root的左儿子当作起点，去判断时候包含Tree2
            result=HasSubtree(pRoot1->left,pRoot2); 
        if(!result)    //如果还找不到，那么就再去root的右儿子当作起点，去判断时候包含Tree2
            result=HasSubtree(pRoot1->right,pRoot2);
        return result; //返回结果
    }
     
    bool isSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(!pRoot2)
            return true; //如果Tree2已经遍历完了都能对应的上，返回true
        if(!pRoot1)
            return false; //如果Tree2还没有遍历完，Tree1却遍历完了。返回false
        bool result=true;
        if(pRoot1->val!=pRoot2->val)//如果其中有一个点没有对应上，返回false
            result=false;
        if(result)    //如果根节点对应的上，那么就去左子节点里面匹配
            result=isSubtree(pRoot1->left,pRoot2->left);
        if(result)    //左子节没对应上，去右子节点里面匹配
            result=isSubtree(pRoot1->right,pRoot2->right);
        return result;    //返回结果
    }
};