1. 程式人生 > 其它 >10-2 相同的樹(leecode 100)

10-2 相同的樹(leecode 100)

技術標籤:# 資料結構與演算法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); } };