16 树的子结构(这题多复习)

Posted 2020-10-16 zqlucky

题目描述

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

 

这题分两步：

　　第1：在树A中找到和树B的根节点的值一样的节点R，注意树的节点值可以有多个相同的值。

　　第2：判断树A中以R为根节点的子树是不是包含和B一样的树结构。

这题难在递归基和判断树节点为空该返回什么，一定要记清楚找头结点的时候遇到空节点就返回错，helper函数如果子树为空，还有就是result的使用。

说明已经匹配完了，返回true；如果是树A为空，说明匹配不到子树，返回false。

 

理解难点：helper每次返回以该根节点开始的是否匹配，然后判断左子树和右子树是否匹配。

 

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