1. 程式人生 > >LeetCode演算法題100:相同的樹解析

LeetCode演算法題100:相同的樹解析

給定兩個二叉樹,編寫一個函式來檢驗它們是否相同。 如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。 示例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))