剑指offer题目17:树的子结构

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

解答思路

画图出来先分析思路。

  1. 先判断两者的根是否相同
  2. 相同则判断两者的左右子树
  3. 不相同则用父树的左子树去和子树比对
  4. 如果子树已经到达终点,则结束比对

分析完了会觉得不算太难,觉得就是难在判断边界条件,注意某一方为NULL的时候。

实现代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result = false;
        
        if(pRoot1 != NULL && pRoot2 != NULL) {
            // 如果两者根相等,判断左右子树
            if(pRoot1->val == pRoot2->val) {
                result = ConfirmSubtree(pRoot1, pRoot2);
            } 
            
            // 如果不相等,则用父树的左树和子树比较
            if(!result) {
                result = HasSubtree(pRoot1->left, pRoot2);
            }
            
            // 如果不相等,则用父树的右树和子树比较
            if(!result) {
                result = HasSubtree(pRoot1->right, pRoot2);
            }
        }

        return result;
        
    }
    
    // 在知道根节点相同的情况下,判断左右子树。
    // 递归的结束条件是两者都到达了最底部。
    bool ConfirmSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {![](https://img2018.cnblogs.com/blog/799055/201904/799055-20190401114937819-449021032.png)

        // 先判断子树是否为NULL,这是因为如果子树和父树都为NULL的话,其实就是子树包含在父树里面
        if(pRoot2 == NULL) {
            return true;
        }
        
        if(pRoot1 == NULL) {
            return false;
        }

        if(pRoot1->val != pRoot2->val) {
            return false;
        } 

        return ConfirmSubtree(pRoot1->left, pRoot2->left) && ConfirmSubtree(pRoot1->right, pRoot2->right);
    }

};
posted @ 2019-04-01 11:55  {-)大傻逼  阅读(149)  评论(0)    收藏  举报
欢迎转载,转载请注明本文地址。