1. 程式人生 > 其它 >線段樹的分裂與合併

線段樹的分裂與合併

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