1. 程式人生 > 其它 >LeetCode 563——二叉樹的坡度(DFS)

LeetCode 563——二叉樹的坡度(DFS)

技術標籤: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;
    }
};

四、解題結果