劍指26 樹的子結構
阿新 • • 發佈:2020-07-04
輸入兩棵二叉樹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 };