二叉树中的最大路径和
题目描述
给定一个非空二叉树,返回器最大路径和,路径指一条从任意节点出发,到达任意节点的序列,该路径至少包含一个节点,且不一定经过根节点
解题思路
树这一类数据结构我还不是很熟悉,需要更进一步的学习,以下思路来自于题解:
根据题意可知,一条最大的路径存在两种可能:
- 存在一个节点,一条最大路径等于该节点的左右子树中路径较大的一颗子树,加上它自己后向其父节点回溯
- 存在一个节点,一条最大路径包含其左右子树与其本身,不会再向父节点回溯
代码实现
package algorithm.tree;
public class MaxRoadLength {
private int num = 0;
public int maxPathSum(TreeNode root) {
num = root.val;
calculate(root);
return num;
}
private int calculate(TreeNode root){
if(root == null){
return 0;
}
int leftLength = Math.max(0,calculate(root.left)); //递归获取左子树中的最大路径
int rightLength = Math.max(0,calculate(root.right)); //递归获取右子树中的最大路径
num = Math.max(num,root.val+leftLength+rightLength); //计算第二种情况下的路径长度,即左子树长度+右子树长度+节点本身的长度
return Math.max(leftLength,rightLength)+root.val;//计算第一种情况下的路径长度,去左右子树中路径长的加上节点本身并向父节点回溯
}
}