LeetCode演算法題100:相同的樹解析
阿新 • • 發佈:2018-12-18
給定兩個二叉樹,編寫一個函式來檢驗它們是否相同。 如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。 示例1:
輸入: 1 1
/ \ / \
2 3 2 3
[1,2,3], [1,2,3]
輸出: true
示例2:
輸入: 1 1
/ \
2 2
[1,2], [1,null,2]
輸出: false
示例3:
輸入: 1 1 / \ / \ 2 1 1 2 [1,2,1], [1,1,2] 輸出: false
之前對樹的瞭解較少,也沒怎麼寫過關於樹的演算法程式等,所以參考了別人的程式,總結一下就是樹應該是要使用遞迴進行遍歷的,當初學習的時候記得遍歷方法好像不止一種,以後隨學隨記,這次的程式就是利用遞迴的思想,如果不是空樹節點的話,就看此節點的值是否相同,然後遞迴比較此節點的左右子節點是否分別相同,都相同就返回真。 C++原始碼:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if (p==NULL || q==NULL)
if (p == q)
return true;
else
return false;
return (p->val==q->val && isSameTree(p->left, q->left) && isSameTree (p->right, q->right));
}
};
python3原始碼:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def isSameTree(self, p, q):
"""
:type p: TreeNode
:type q: TreeNode
:rtype: bool
"""
if p==None or q==None:
if p==q:
return True
else:
return False
return (p.val==q.val and self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right))