题解 | #农场最大产奶牛群#
农场最大产奶牛群
https://www.nowcoder.com/practice/16d827f124e14e05b988f3002e7cd651
农场最大产奶牛群
1.考虑树上每个点存放左右子树的较大值+自身的val
2.考虑计算ans的时候需要取左边子树的最大值+右边子树的最大值+自身val
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
int dfs(TreeNode* root,int* ans) {
if (!root) return 0;
int x1 = dfs(root->left, ans);
int x2 = dfs(root->right, ans);
*ans = max(*ans, x1 + x2 + root->val);
return max(x1, x2) + root->val;
}
int maxMilkSum(TreeNode* root) {
int ans = 0;
dfs(root, &ans);
return ans;
}
};