10-2 相同的樹(leecode 100)
阿新 • • 發佈:2021-02-10
技術標籤:# 資料結構與演算法06-樹
1 問題
給你兩棵二叉樹的根節點 p 和 q ,編寫一個函式來檢驗這兩棵樹是否相同。
如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。
輸入:p = [1,2,3], q = [1,2,3]
輸出:true
2 解法
/**
* 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 compare(TreeNode* a, TreeNode* b)
{
//2.遞迴結束條件
//a,b都為空
if(a == nullptr && b == nullptr)
return true;
//a為空,b不為空
else if(a == nullptr && b != nullptr)
return false;
//a不為空,b為空
else if(a != nullptr && b == nullptr)
return false;
//a,b都不為空,但值不等
else if(a->val != b->val)
return false;
//3.每層遞迴的邏輯
//a,b都不為空,且值相等
else
{
//比較a,b的左側
bool left = compare(a->left, b->left);
//比較a,b的右側
bool right = compare(a->right, b->right);
return left && right;
}
}
bool isSameTree(TreeNode* p, TreeNode* q) {
return compare(p, q);
}
};