LeetCode——563. 二叉樹的坡度(Java)
阿新 • • 發佈:2021-11-18
題目描述
題幹: 給定一個二叉樹,計算 整個樹 的坡度 。 一個樹的 節點的坡度 定義即為,該節點左子樹的節點之和和右子樹節點之和的 差的絕對值 。 如果沒有左子樹的話,左子樹的節點之和為 0 ;沒有右子樹的話也是一樣。空結點的坡度是 0 。 整個樹 的坡度就是其所有節點的坡度之和。 示例 1: 輸入:root = [1,2,3] 輸出:1 解釋: 節點 2 的坡度:|0-0| = 0(沒有子節點) 節點 3 的坡度:|0-0| = 0(沒有子節點) 節點 1 的坡度:|2-3| = 1(左子樹就是左子節點,所以和是 2 ;右子樹就是右子節點,所以和是 3 ) 坡度總和:0 + 0 + 1 = 1 示例 2: 輸入: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 ,和是 16 ) 坡度總和:0 + 0 + 0 + 2 + 7 + 6 = 15 示例 3: 輸入:root = [21,7,14,1,1,2,2,3,3] 輸出:9
題解思路
返回二叉樹的坡度,首先要明白坡地的定義,即節點左右的子樹節點之和的差的絕對值
顯然我們只需要遍歷完二叉樹,遍歷中將每個節點的坡度計算然後累加即可
這裡採用深度優先遍歷,如果節點為葉子節點,則返回0,其他則返回左右節點和即可
正確程式碼
class findTiltFindTiltSolution { private int slope = 0; public int findTilt(TreeNode root) { dfs(root); return slope; } private int dfs(TreeNode root) { if (root == null) { return 0; } int sumLeft = dfs(root.left); int sumRight = dfs(root.right); slope += Math.abs(sumLeft - sumRight); return sumLeft + root.val + sumRight; } }
總結
深度優先遍歷的時間和空間複雜度都為為O(n),都相當於只遍歷一遍二叉樹
如果文章存在問題或者有更好的題解,歡迎在評論區斧正和評論,各自努力,最高處見