【leetcode】【medium】剑指 Offer 26. 树的子结构​​​​​​​

剑指 Offer 26. 树的子结构

输入两棵二叉树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

题目链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof/

思路

两层dfs嵌套。

第一层：负责对A所有节点作为根的遍历。

第二层：确定A上的根之后，确定是否存在子树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool match(TreeNode* A, TreeNode* B){
        if(!B){
            return true;
        }
        if(!A){
            return false;
        }
        if(A->val == B->val){
            if(match(A->left, B->left) && match(A->right, B->right)){
                return true;
            }
        }
        return false;
    }
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if(!B){
            return false;
        }
        if(match(A, B)){
            return true;
        }
        if((A->left && isSubStructure(A->left, B)) || (A->right && isSubStructure(A->right, B))){
            return true;
        }
        return false;
    }
};