1. 程式人生 > 其它 >LeetCode50--二叉樹的坡度、另一個樹的子樹

LeetCode50--二叉樹的坡度、另一個樹的子樹

技術標籤:Leetcode刷題# Leetcode經典好題

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);
		}
	}