剑指Offer——树的子结构

  本博客为刷题笔记，方便日后复习。
  刷题平台：牛客网

题目描述

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

题目分析

• 子树定义：
    树A和树B的根结点相等，并且树A的左子树和树B的左子树相等，树A的右子树和树B的右子树相等
• 分析
    要查找树A中是否存在和数B结构一样的子树，我们可以分为两步，第一步在树A中找到和树B的根节点的值一样的节点R，第二步判断树A中以R为根节点的子树是否包含和树B一样的结构。
    这就是类似于数的深度优先遍历一样，使用递归的方法就可求解。
• 递归分析：
  第一步：
    根据题意可知，需要一个函数判断树A和树B是否有相同的结构。显然是个递归程序。可考察递归程序3部曲。
    递归函数的功能：判断2个数是否有相同的结构，如果相同，返回true，否则返回false
  递归终止条件：
    如果树B为空，返回true，此时，不管树A是否为空，都为true
    否则，如果树B不为空，但是树A为空，返回false，此时B还没空但A空了，显然false
  下一步递归参数：
    如果A的根节点和B的根节点不相等，直接返回false
    否则，相等，就继续判断A的左子树和B的左子树，A的右子树和B的右子树
  第二步：
    有了上面那个函数，接下来就应该让树A的每个节点作为根节点来和B树进行比较。
    遍历树A的每个节点，可用遍历算法。这里采用先序遍历。

代码实现

/*
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)
    {
        if(!pRoot1)
            return false;
        if(!pRoot2)
            return false;
        return ( dfs(pRoot1,pRoot2)) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
    }
// 定义的深度优先遍历方法
private:
    bool dfs(TreeNode * r1, TreeNode * r2){
        if(!r2) 
            return true;
        if(!r1)
            return false;
        if(r1->val != r2->val)
            return false;
        return dfs(r1->left, r2->left) && dfs(r1->right, r2->right);
    }
};