LeetCode 563——二叉樹的坡度(DFS)
阿新 • • 發佈:2021-02-14
技術標籤:LeetCode刷題
一、題目介紹
給定一個二叉樹,計算 整個樹 的坡度 。
一個樹的 節點的坡度 定義即為,該節點左子樹的節點之和和右子樹節點之和的 差的絕對值 。如果沒有左子樹的話,左子樹的節點之和為 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
輸入: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
輸入:root = [21,7,14,1,1,2,2,3,3]
輸出:9
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/binary-tree-tilt
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
二、解題思路
本題通過DFS的方法計算每個節點的左右子樹之和,並在遞迴過程中計算每個節點坡度值再進行求和。詳見程式碼。
三、解題程式碼
class Solution { public: int treeTilt = 0; int findTilt(TreeNode* root) { solve(root); return treeTilt; } int solve(TreeNode* node) { if(node == NULL) return 0; int Left = solve(node->left); //計算左子樹的節點值之和 int Right = solve(node->right); //計算右子樹的節點值之和 treeTilt += abs(Left - Right); return Left + Right + node->val; } };