1. 程式人生 > 其它 >leetcode-華為專題-572. 另一棵樹的子樹

leetcode-華為專題-572. 另一棵樹的子樹

/**
 * 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) { if(root==NULL||subRoot==NULL) return false; if(root->val!=subRoot->val){ return isSubtree(root->left,subRoot)||isSubtree(root->right,subRoot); }else{
// return judge(root,subRoot); return judge(root,subRoot)||isSubtree(root->left,subRoot)||isSubtree(root->right,subRoot); // 這是為了應對出現有重複元素出現的情況,比如root: 1 1, subroot:1 } return false; } bool judge(TreeNode* root1, TreeNode* root2){ // 這是用來判斷子樹的 if(root1==NULL&&root2==NULL) //
兩個都為空,說明比對完了,返回true return true; if(root1==NULL||root2==NULL) // 其中一個為空,說明一個比完,一個還有,返回false return false; // 如果是判斷子結構的就用下面這種方式 // if(root2==NULL) // return true; // if(root1==NULL) // 此時肯定root2不為null,root1為null,所以false; // return false; if(root1->val!=root2->val) return false; else return judge(root1->left,root2->left)&&judge(root1->right, root2->right); } };