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

LeetCode——563. 二叉樹的坡度(Java)

題目描述

題幹:
給定一個二叉樹,計算 整個樹 的坡度 。
一個樹的 節點的坡度 定義即為,該節點左子樹的節點之和和右子樹節點之和的 差的絕對值 。
如果沒有左子樹的話,左子樹的節點之和為 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),都相當於只遍歷一遍二叉樹

如果文章存在問題或者有更好的題解,歡迎在評論區斧正和評論,各自努力,最高處見