LeetCode50--二叉樹的坡度、另一個樹的子樹
阿新 • • 發佈:2021-02-12
1.二叉樹的坡度
//給定一個二叉樹,計算 整個樹 的坡度 。 // // 一個樹的 節點的坡度 定義即為,該節點左子樹的節點之和和右子樹節點之和的 差的絕對值 。如果沒有左子樹的話,左子樹的節點之和為 0 ;沒有右子樹的話也是一 //樣。空結點的坡度是 0 。 // // 整個樹 的坡度就是其所有節點的坡度之和。 // // // // 示例 1: // 1 1 // / \ -> / \ // 2 3 0 0 // //輸入:root = [1,2,3] //輸出:1 //解釋: //節點 2 的坡度:|0-0| = 0(沒有子節點) //節點 3 的坡度:|0-0| = 0(沒有子節點) //節點 1 的坡度:|2-3| = 1(左子樹就是左子節點,所以和是 2 ;右子樹就是右子節點,所以和是 3 ) //坡度總和:0 + 0 + 1 = 1 // // // 示例 2: // 4 6 // / \ / \ // 2 9 -> 2 7 // / \ \ / \ \ // 3 5 7 0 5 0 //輸入:root = [4,2,9,3,5,null,7] //輸出:15 //解釋: //節點 3 的坡度:|0-0| = 0(沒有子節點) //節點 5 的坡度:|0-0| = 0(沒有子節點) //節點 7 的坡度:|0-0| = 0(沒有子節點) //節點 2 的坡度:|3-5| = 2(左子樹就是左子節點,所以和是 3 ;右子樹就是右子節點,所以和是 5 ) //節點 9 的坡度:|0-7| = 7(沒有左子樹,所以和是 0 ;右子樹正好是右子節點,所以和是 7 ) //節點 4 的坡度:|(3+5+2)-(9+7)| = |10-16| = 6(左子樹值為 3、5 和 2 ,和是 10 ;右子樹值為 9 和 7 ,和是 1 //6 ) //坡度總和:0 + 0 + 0 + 2 + 7 + 6 = 15 // // // 示例 3: // 21 3 // / \ / \ // 7 14 6 0 // / \ /\ -> / \ / \ // 1 1 2 2 0 0 0 0 // / \ / \ // 3 3 0 0 //輸入:root = [21,7,14,1,1,2,2,3,3] //輸出:9 // // // // // 提示: // // // 樹中節點數目的範圍在 [0, 104] 內 // -1000 <= Node.val <= 1000 // // Related Topics 樹 深度優先搜尋 遞迴
還是用一般的方法,定義一個成員變數,專門用於做差值的累加,定義一個函式專門計算以該節點為根節點的子樹的所有節點之和,返回的是節點之和,但是同時也在該函式中做累加。
class Solution { int ans = 0; public int findTilt(TreeNode root) { helper(root); return ans; } public int helper(TreeNode root){ if(root == null){ return 0; } int left_sum = helper(root.left); int right_sum = helper(root.right); ans += Math.abs(left_sum - right_sum); return left_sum + right_sum + root.val; } }
2.另一個樹的子樹
//給定兩個非空二叉樹 s 和 t,檢驗 s 中是否包含和 t 具有相同結構和節點值的子樹。s 的一個子樹包括 s 的一個節點和這個節點的所有子孫。s 也可以看 //做它自身的一棵子樹。 // // 示例 1: //給定的樹 s: // // // 3 // / \ // 4 5 // / \ // 1 2 // // // 給定的樹 t: // // // 4 // / \ // 1 2 // // // 返回 true,因為 t 與 s 的一個子樹擁有相同的結構和節點值。 // // 示例 2: //給定的樹 s: // // // 3 // / \ // 4 5 // / \ // 1 2 // / // 0 // // // 給定的樹 t: // // // 4 // / \ // 1 2 // // // 返回 false。 // Related Topics 樹
這裡主要需要看一棵樹是否為另一棵樹的子樹與兩棵樹是否相等,這兩種情況的判斷條件:
一個是或一個是與。
class Solution {
public boolean isSubtree(TreeNode s, TreeNode t) {
return dfs(s, t);
}
public boolean dfs(TreeNode s, TreeNode t) {
if (s == null && t == null) {
return true;
} else if (s == null || t == null) {
return false;
}
return check(s, t) || dfs(s.left, t) || dfs(s.right, t);
}
public boolean check(TreeNode s, TreeNode t) {
if (s == null && t == null) {
return true;
} else if (s == null || t == null) {
return false;
}
return s.val == t.val && check(s.left, t.left) && check(s.right, t.right);
}
}