ツリーの下部構造-サブツリーをトラバースします

木の下部構造

2つの二分木AとBを入力して、BがAの下位構造であるかどうかを判別します。（空のツリーはどのツリーの下位構造でもないことに同意します）
BはAの下位構造です。つまり、AはBと同じ構造とノード値を持ちます。

例：
与えられたツリーA：

     3
    / \
   4   5
  / \
 1   2

与えられたツリーB：

   4 
  /
 1

BとAのサブツリーは同じ構造とノード値を持っているため、trueを返します。

例1：
入力：A = [1,2,3]、B = [3,1]
出力：false

例2：
入力：A = [3,4,5,1,2]、B = [4,1]
出力：真の
制限：

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;
}