1. 程式人生 > 實用技巧 >劍指26 樹的子結構

劍指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

遞迴求解,關鍵還是檢查程式碼的魯棒性。

做題時忘記對B==nullptr做判斷,只在checksame裡面做了判斷,但是沒有注意findbegin也需要做判斷。

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     bool isSubStructure(TreeNode* A, TreeNode* B) {
13 if(B==nullptr) 14 return false; 15 return findbegin(A,B); 16 } 17 18 bool findbegin(TreeNode* A, TreeNode* B){ 19 if(A==nullptr) 20 return false; 21 return ((A->val==B->val)&&checksame(A,B)) 22 |findbegin(A->left,B)|findbegin(A->right,B);
23 } 24 25 bool checksame(TreeNode* base, TreeNode* sub){ 26 if(base==nullptr && sub==nullptr) 27 return true; 28 if(base==nullptr) 29 return false; 30 if(sub==nullptr) 31 return true; 32 if(base->val!=sub->val) 33 return false; 34 return true&checksame(base->left,sub->left)&checksame(base->right,sub->right); 35 } 36 };