563. 二叉樹的坡度
給定一個二叉樹,計算整個樹的坡度。
一個樹的節點的坡度定義即爲,該節點左子樹的結點之和和右子樹結點之和的差的絕對值。空結點的的坡度是0。
整個樹的坡度就是其所有節點的坡度之和。
示例:
輸入: 1 / \ 2 3 輸出: 1 解釋: 結點的坡度 2 : 0 結點的坡度 3 : 0 結點的坡度 1 : |2-3| = 1 樹的坡度 : 0 + 0 + 1 = 1注意:
- 任何子樹的結點的和不會超過32位整數的範圍。
- 坡度的值不會超過32位整數的範圍。
解法一
//時間複雜度O(n), 空間複雜度O(1)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int getTilt(TreeNode* root, int& tilt) {
if(!root) return 0;
int suml = getTilt(root->left, tilt);
int sumr = getTilt(root->right, tilt);
tilt += abs(suml - sumr);
return suml + sumr + root->val;
}
int findTilt(TreeNode* root) {
int tilt = 0;
getTilt(root, tilt);
return tilt;
}
};
思路:
後序遍歷,函數getTilt返回值爲樹結點值的和,求和過程中對變量tilt作累計求和,遍歷完成後返回tilt。