剑指 Offer 26. 树的子结构

题目描述

思路分析

利用两个递归，第一个先序遍历递归访问A所有节点，第二个递归，比较是否相等。
第一个递归进行了优化，遇到节点值相等进第二个递归，第二个递归，B走到空就说明包含。

代码展示

/**
 * 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 isSubStructure(TreeNode* A, TreeNode* B) {
        if (A == NULL || B == NULL) {
            return false;
        }
        bool ret;
        if (A->val == B->val){
            ret = includeSametree(A, B);
        }
        return ret || isSubStructure(A->left, B) || isSubStructure(A->right, B);
    }
    
    bool includeSametree(TreeNode* A, TreeNode* B) {
        if(B == NULL) {
            return true;
        }
        if (A == NULL || A->val != B->val) {
            return false;
        }
        return includeSametree(A->left, B->left) && includeSametree(A->right, B->right);
    }
};

结果分析

时间复杂度 O(MN)
空间复杂度 O(N)