剑指offer26.树的子结构

题目描述

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B：

   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true
限制：

0 <= 节点个数 <= 10000

思路

双重递归：外层递归前序遍历树A，内层递归对A中的每一个节点都作为root和树B进行匹配。

我的代码：

class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if (!A || !B) return false;
        bool ismatch;
        if (A->val == B->val) {
            ismatch = isMatched(A, B);
        }
        if (ismatch) return true;
        
        return isSubStructure(A->left, B) || isSubStructure(A->right, B);
    }
        
    bool isMatched(TreeNode *A, TreeNode *B) {
        //如果都走到空说明之前的val都相等，或者A不为空，B为空，说明B所有节点已遍历完成
        if (A == NULL && B == NULL || B == NULL) return true;
        if (A == NULL && B) return false;
        if (A->val != B->val) return false;
        return isMatched(A->left, B->left) && isMatched(A->right, B->right);
    }
};

题解高赞的代码：

bool isMatched(TreeNode *A, TreeNode *B) {
    //B为空，说明B所有节点已遍历完成
    if (B == NULL) return true;
    //A为空或A B结点值不等，说明匹配失败
    if (A == NULL || A->val != B->val) return false;
    return isMatched(A->left, B->left) && isMatched(A->right, B->right);
}
//每个节点都尝试作为根
bool isSubStructure(TreeNode* A, TreeNode* B) {
    return (A != NULL && B != NULL) && (isMatched(A, B) 
    || isSubStructure(A->left, B) || isSubStructure(A->right, B));
}