17、剑指offer--树的子结构

题目描述

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

 

解题思路：

1、在1中找到与2根节点相同的结点R

2、判断1中以R为根节点的子树是否包含2

3、如果不包含，再在1的左子树、右子树分别找，直到全部找完

#include <iostream>
#include <malloc.h>
using namespace std;
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
class Solution {
public:
    //1、在1中找到与2根节点相同的结点R
    //2、判断1中以R为根节点的子树是否包含2
    //3、如果不包含，再在1的左子树、右子树分别找，知道全部找完
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result = false;
        if(pRoot1 != NULL && pRoot2 != NULL)
        {
            if(pRoot1->val == pRoot2->val)
            {
                result = Tree1HasTree2(pRoot1,pRoot2);
            }
            if(!result)
            {
                result = HasSubtree(pRoot1->left,pRoot2);
            }
            if(!result)
            {
                result = HasSubtree(pRoot1->right,pRoot2);
            }
        }
        return result;
    }
    bool Tree1HasTree2(TreeNode *pRoot1,TreeNode *pRoot2)
    {
        if(pRoot2 == NULL)
            return true;
        if(pRoot1 == NULL)
            return false;
        if(pRoot1->val != pRoot2->val)
            return false;
        //1的左右子树是否和2的左右子树匹配
        return Tree1HasTree2(pRoot1->left,pRoot2->left)&&Tree1HasTree2(pRoot1->right,pRoot2->right);
    }
};
//先序建立二叉树
TreeNode *CreateBiTree(){
    int ch;

    TreeNode *T;
    cin>>ch;
    if(ch== 0)T=NULL;
    else{
        T = (TreeNode *)malloc(sizeof(TreeNode));
        T->val = ch;
        cout<<"输入左子节点"<<endl;
        T->left = CreateBiTree();
        cout<<"输入右子节点"<<endl;
        T->right = CreateBiTree();
    }
    return T;//返回根节点
}
//先序遍历二叉树
void PreOrderTraverse(TreeNode *T){
    if(T)
    {
       cout<<T->val<<" ";
       PreOrderTraverse(T->left);
       PreOrderTraverse(T->right);
    }
}
int main()
{
    TreeNode *T1;
    cout<<"输入根节点："<<endl;
    T1 = CreateBiTree();//建立
    cout<<"树1的先序遍历为"<<endl;
    PreOrderTraverse(T1);//输出
    cout<<endl;
    cout<<"输入根节点："<<endl;
    TreeNode *T2;
    T2 = CreateBiTree();//建立
    cout<<"树2的先序遍历为"<<endl;
    PreOrderTraverse(T2);//输出
    cout<<endl;
    Solution s;
    bool res;
    res = s.HasSubtree(T1,T2);
    if(res)
        cout<<"树2是树1的子树"<<endl;
    else
        cout<<"树2不是树1的子树"<<endl;
    return 0;
}