剑指offer——普通二叉树的子结构

1.题目描述

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

2.解题思路

  首先要进行程序的鲁棒性分析：
1. 如果B为空树，那么判断结果为false，因为空树不是任意一棵树的子结构；
2. 如果A为空树，那么判断结果也为false。
  如此分析以后，只需对A与B均不是空树的情况进行分析。分析思路如下：
1）判断当前A节点和B树的根节点是否相等
2）如果当前节点相等，那么继续对二者当前节点的左右子树进行判断。如果左右子树不等，那么就不是子结构。
3）如果步骤1）中二者当前节点不等，那么则在A中选择左子树作为新的根节点进行步骤1）。
4）如果在左子树中未能判定为子结构，那么则挑选A的右孩子作为新的根节点进行判断。

3.解答——cpp

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    // help
    bool IsSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if (pRoot2 == nullptr){
            return true;
        }
        if (pRoot1 == nullptr && pRoot2 != nullptr){
            return false;
        }
        if (pRoot1->val != pRoot2->val){
            return false;
        }
        return (IsSubtree(pRoot1->left,pRoot2->left) && IsSubtree(pRoot1->right,pRoot2->right));

    }

    // Original and Robuness
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result = false;
        // 空树不是任意一个树的子结构，即B为空树时直接返回false
        if (pRoot2 == nullptr)
            return false;
        // 当A为空树且B非空时，B不是A的子结构
        if (pRoot1 == nullptr && pRoot2 != nullptr)
            return false;
        // 当A、B都不是空树时
        // 调用IsSubtree判定B是不是A的子结构
        // 如果当前指向的节点相等，则调用Is继续判断
        if (pRoot1->val == pRoot2->val){
            result = IsSubtree(pRoot1,pRoot2);
        }
        // 如果当前节点判定失败，则将当前节点的左子树作为新的根节点重新判断
        if (!result){
            result = HasSubtree(pRoot1->left,pRoot2);
        }
        // 如果左子树判定也失败，则选择右子树判定
        if (!result){
            result = HasSubtree(pRoot1->right,pRoot2);
        }
        // 返回最终结果
        return result;
    }
};