LeetCode中如何解決相同的樹問題

題目鏈接

https://leetcode-cn.com/problems/same-tree/

題目描述

給定兩個二叉樹，編寫一個函數(shù)來檢驗它們是否相同。

如果兩個樹在結(jié)構(gòu)上相同，并且節(jié)點具有相同的值，則認為它們是相同的。

示例 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

解題方案

思路

• 標簽：深度優(yōu)先遍歷

• 終止條件與返回值：

• 當兩棵樹的當前節(jié)點都為null時返回true

• 當其中一個為null另一個不為null時返回false

• 當兩個都不為空但是值不相等時，返回false

• 執(zhí)行過程：當滿足終止條件時進行返回，不滿足時分別判斷左子樹和右子樹是否相同，其中要注意代碼中的短路效應

• 時間復雜度：O(n)，n為樹的節(jié)點個數(shù)

代碼

• Java版本

/**
* Definition for a binary tree node.
* public class TreeNode {
*     int val;
*     TreeNode left;
*     TreeNode right;
*     TreeNode(int x) { val = x; }
* }
*/
class Solution {
   public boolean isSameTree(TreeNode p, TreeNode q) {
       if(p == null && q == null)
           return true;
       if(p == null || q == null)
           return false;
       if(p.val != q.val)
           return false;
       return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
   }
}

• JavaScript版本

/**
* Definition for a binary tree node.
* function TreeNode(val) {
*     this.val = val;
*     this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} p
* @param {TreeNode} q
* @return {boolean}
*/
var isSameTree = function(p, q) {
   if(p == null && q == null)
       return true;
   if(p == null || q == null)
       return false;
   if(p.val != q.val)
       return false;
   return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
};

畫解