572. 另一棵樹的子樹
阿新 • • 發佈:2021-12-08
題目連結:
給你兩棵二叉樹 root
和 subRoot
。檢驗 root
中是否包含和 subRoot
具有相同結構和節點值的子樹。如果存在,返回 true
;否則,返回 false
。
二叉樹 tree
的一棵子樹包括 tree
的某個節點和這個節點的所有後代節點。tree
也可以看做它自身的一棵子樹。
示例 1:
輸入:root = [3,4,5,1,2], subRoot = [4,1,2]
輸出:true
示例 2:
輸入:root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
輸出:false
題解
思路:該題與
程式碼(C++):
class Solution { public: bool isSameTree(TreeNode* p, TreeNode* q) { //判斷兩棵樹是否相等if (p != nullptr && q == nullptr) return false; else if (p == nullptr && q != nullptr) return false; else if (p == nullptr && q == nullptr) return true; else if (p->val != q->val) return false; bool left = isSameTree(p->left, q->left); //1號樹:左、 2號樹:左 bool right = isSameTree(p->right, q->right); // 1號樹:右、 2號樹:右 return left && right; // 1號樹:中、 2號樹:中 (邏輯處理) } bool isSubtree(TreeNode* root, TreeNode* subRoot) { if (subRoot == nullptr) return true; if (root == nullptr) return false; //判斷 一棵樹 是否為 另一棵樹 的子樹 bool result1 = isSameTree(root, subRoot); bool result2 = isSubtree(root->left, subRoot); bool result3 = isSubtree(root->right, subRoot); return result1 || result2 || result3; } };
程式碼(JavaScript):
//遞迴法 var isSameTree = function(p, q) { if (p === null && q === null) return true; else if (p === null || q === null) return false; else if (p.val != q.val) return false; var result1 = isSameTree(p.left, q.left); var result2 = isSameTree(p.right, q.right); return result1 && result2; }; var isSubtree = function(root, subRoot) { if (root === null && subRoot === null) return true; else if (root === null && subRoot != null) return false; var result1 = isSameTree(root, subRoot); var result2 = isSubtree(root.left, subRoot); var result3 = isSubtree(root.right, subRoot); return result1 || result2 || result3; };