树的子结构

Posted 2021-06-27 vector6_

树的子结构

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

Input: {8,8,#,9,#,2,#,5},{8,9,#,2}

Return: true

子树和子结构的区别:

子树的意思是包含了一个结点，就得包含这个结点下的所有节点，一棵大小为n的二叉树有n个子树，就是分别以每个结点为根的子树。子结构的意思是包含了一个结点，可以只取左子树或者右子树，或者都不取。

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    //先在大树中找到与小树根节点相同的节点，然后再以此节点为根节点，对比大树和小树的左右子节点是否相同
    bool HasSubtreeHelper(TreeNode* p1, TreeNode* p2)
    {
        if(p2==nullptr)                     //如果小树先到达空节点，小树为空也是子结构
            return true;					
        if(p1==nullptr)						//如果大树到达空节点，但是由上行判断小树不为空节点，此时返回false
            return false;
        if(p1->val != p2->val)
            return false;
        return HasSubtreeHelper(p1->left, p2->left) && HasSubtreeHelper(p1->right, p2->right);
    }
    
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
        if(pRoot1==nullptr || pRoot2==nullptr)
            return false;
        bool result = false;
        if(pRoot1->val == pRoot2->val)       //先在大树中找到与小树根节点相同的节点，然后再以此节点为根节点，对比大树和小树的左右子节点是否相同
        {
            result = HasSubtreeHelper(pRoot1, pRoot2);
        }
        if(!result)
            result = HasSubtree(pRoot1->left, pRoot2);
        if(!result)
            result = HasSubtree(pRoot1->right, pRoot2);
        return result;
    }
};