树的子结构 - 遍历子树

树的子结构

输入两棵二叉树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的所有子树，拿来与B树作比较就可以了。

解法

bool aha(struct TreeNode* A, struct TreeNode* B){
    // 如果A节点为空，B不为空，说明此时已经不匹配，返回false
    if(A == NULL && B){
        return false;
    }
    // 如果B为空，说明当前节点是空节点，从某处到此的一个分支已经完成了，如果没有false出现，说明这个子树是A的子树
    if(B == NULL){
        return true;
    }
    // A与B不等，且B不为空
    if(A->val != B->val && B != NULL)
        // 此时应该return false，因为当前节点不一定是原始B的根节点，所以不能继续向下判断
        return false;
    // 如果AB相同，此时可以往下遍历，确保AB的左右子树都相同
    if(A->val == B->val){
        return aha(A->left, B->left) && aha(A->right, B->right);
    }
    return false;
}

void dfs_A(struct TreeNode* A, struct TreeNode* B, bool *flag){
    // 如果A为空了，不继续往下遍历
    if(A == NULL){
        return;
    }
    // A子树与B比较
    if(aha(A, B)){
        // A, B匹配成功
        *flag = true;
        return;
    }
    else{
        // 遍历A树，
        dfs_A(A->left, B, flag);
        dfs_A(A->right, B, flag);
    }
}

bool isSubStructure(struct TreeNode* A, struct TreeNode* B){
    //代码的鲁棒性，空树，算吗？
    if(A == NULL && B == NULL){
        return false;
    }
    // B为NULL，返回false
    if(B == NULL)
        return false;
    bool *flag = malloc(sizeof(bool));
    *flag = false;
    dfs_A(A,B,flag);
    return *flag;
}