Skip to main content

572. Subtree of Another Tree

Given the roots of two binary trees root and subRoot, return true if there is a subtree of root with the same structure and node values of subRoot and false otherwise.

A subtree of a binary tree tree is a tree that consists of a node in tree and all of this node's descendants. The tree tree could also be considered as a subtree of itself.

Example 1:

image

Input: root = [3,4,5,1,2], subRoot = [4,1,2] Output: true

Example 2:

image

Input: root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2] Output: false

Constraints:

  • The number of nodes in the root tree is in the range [1, 2000].
  • The number of nodes in the subRoot tree is in the range [1, 1000].
  • -104 <= root.val <= 104
  • -104 <= subRoot.val <= 104
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool isSubtree(TreeNode* root, TreeNode* subRoot) {
// base case
if (!subRoot) return true;
if (!root) return false;

// 如果相符就 return true
if (isSame(root, subRoot)) return true;

// 檢查左邊、檢查右邊,只要其中一邊有 subtree 就 return true
return isSubtree(root->left, subRoot) || isSubtree(root->right, subRoot);

}
bool isSame(TreeNode* t1, TreeNode* t2) {
if (!t1 && !t2) return true;
if (!t1 || !t2) return false;
if (t1->val != t2->val) return false;
return isSame(t1->left, t2->left) && isSame(t1->right, t2->right);
}
};
  • T: O(MN)O(M \cdot N)
  • S: O(M+N)O(M + N)