線段樹的分裂與合併
阿新 • • 發佈:2022-05-15
563. 二叉樹的坡度
給你一個二叉樹的根節點 root ,計算並返回 整個樹 的坡度 。
一個樹的 節點的坡度 定義即為,該節點左子樹的節點之和和右子樹節點之和的 差的絕對值 。如果沒有左子樹的話,左子樹的節點之和為 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, 104] 內
-1000 <= Node.val <= 1000
程式碼實現:
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: int findTilt(TreeNode* root) { if(!root) return 0; int res = 0; dfs(root, res); return res; } int dfs(TreeNode *root, int &res) { if(!root) return 0; int l = dfs(root->left, res); int r = dfs(root->right, res); res += abs(l-r); return l + r + root->val; } };