LeetCode 563 二叉树的坡度[dfs] HERODING的LeetCode之路
解题思路:
一道非常简单的题目,不能被绕进去,首先要把题目读明白,左右坡度差本质上是左右子树和之差,那么问题就简单了,每次dfs,返回的就是当前root所在子树之和,更新坡度差的话就是当前root左右子树之差,这两个要分开来计算,代码如下:
/**
* 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 {
private:
int ans = 0;
public:
int findTilt(TreeNode* root) {
dfs(root);
return ans;
}
int dfs(TreeNode* root) {
if(root == nullptr) {
return 0;
}
int sumA = dfs(root -> left);
int sumB = dfs(root -> right);
ans += abs(sumA - sumB);
return sumA + sumB + root -> val;
}
};
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
二维码