leetcode——Binary Tree Maximum Path Sum

    这题很难,主要是我没理解题目的意思,后来看着给的测试参考,和网上找的答案,终于理解了。

例如: 给的

image

正确的路线应该是

 

image

也就是说,路径是一条的,不是有分叉的,之所以算出55是因为把>0的数都加上去了,这样路径就分叉了,就不是一条路径了,所以用dfs做时,返回值应该是左子树或者右子树中>0且比较大的那个子树的路径的值加上当前根节点的值。

我的是参考网上答案的代码:

// 用dfs,因为路径是任意的,所以根节点也是任意的,如果左子树的值>0,即对sum有贡献,则sum加其值,右子树的同理
// 在最后返回时,只能返回一个方向上的值,因为left-->root或者right—>root
class Solution {
private:
	int max_sum;
public:
	int maxPathSum(TreeNode *root) {
		max_sum = INT_MIN;
		dfs(root);
		return max_sum;
	}
	int dfs(TreeNode *root){
		if(root == NULL) return 0;
		int sum = root->val;
		int l = dfs(root->left);
		int r = dfs(root->right);
		if(l > 0) sum += l;
		if(r > 0) sum += r;
		max_sum = max(max_sum, sum);
		return max(l, r) > 0 ? max(l, r) + root->val : root->val;
	}
};
posted @ 2015-01-13 20:56  薇清浅  阅读(146)  评论(0编辑  收藏  举报