java 求二叉树中的最大路径和

 

/**
 * 树节点
 * @author Administrator
 *
 */
public class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) {
          val = x; 
      }
}
 

 

 

    /**
     *  二叉树中的最大路径和
     *  本题中，路径被定义为一条从树中任意节点出发，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。
     */
      int max_sum = Integer.MIN_VALUE;
      public int max_gain(TreeNode node) {
        if (node == null) return 0;

        int left_gain = Math.max(max_gain(node.left), 0);
        int right_gain = Math.max(max_gain(node.right), 0);

        int price_newpath = node.val + left_gain + right_gain;
        max_sum = Math.max(max_sum, price_newpath);

        return node.val + Math.max(left_gain, right_gain);
      }

      public int maxPathSum(TreeNode root) {
        max_gain(root);
        return max_sum;
      }