https://leetcode-cn.com/problems/binary-tree-tilt/
题目
给定一个二叉树,计算整个树的坡度 。
一个树的节点的坡度定义即为,该节点左子树的节点之和和右子树节点之和的差的绝对值。如果没有左子树的话,左子树的节点之和为0;没有右子树的话也是一样。空结点的坡度是0。
整个树的坡度就是其所有节点的坡度之和。
思路1
这个算是非常经典的二叉树递归练习题了,直接对问题的定义建模就好。
首先,求左子树的节点之和,那么有:
int sumLeft = dfs(node->left);
右子树同理:
int sumRight = dfs(node->right);
差的绝对值:
ans += abs(sumLeft - sumRight);
最后有:
int ans = 0;
int findTilt(TreeNode* root) {
dfs(root);
return ans;
}
int dfs(TreeNode* node) {
if (node == nullptr) {
return 0;
}
int sumLeft = dfs(node->left);
int sumRight = dfs(node->right);
ans += abs(sumLeft - sumRight);
return sumLeft + sumRight + node->val;
}
本题就相当于是一个"求二叉树所有节点和"的简单改版。