題目:
Given a binary tree, find the maximum path sum.
The path may start and end at any node in the tree.
For example:
Given the below binary tree,
1 / \ 2 3
Return 6
.
題解:
遞迴求解。
取當前點和左右邊加和,當前點的值中最大的作為本層返回值。並在全域性維護一個max。使用陣列,因為是引用型別。所以在遞迴過程中可以儲存結果。
程式碼如下:
1 public int maxPathSum(TreeNode root) {
2 int[] max = new int[1];
3 max[0] = Integer.MIN_VALUE;
4 findmax(root,max);
5 return max[0];
6 }
7
8 public int findmax(TreeNode root, int[] max){
9 if(root==null)
10 return 0;
11
12 int left = findmax(root.left,max);
13 int right = findmax(root.right,max);
14
15 int ans = Math.max(root.val,Math.max(root.val+left, root.val+right));
16
17 max[0] = Math.max(max[0],Math.max(ans,root.val+left+right));
18
19 return ans;
20
21 }
2 int[] max = new int[1];
3 max[0] = Integer.MIN_VALUE;
4 findmax(root,max);
5 return max[0];
6 }
7
8 public int findmax(TreeNode root, int[] max){
9 if(root==null)
10 return 0;
11
12 int left = findmax(root.left,max);
13 int right = findmax(root.right,max);
14
15 int ans = Math.max(root.val,Math.max(root.val+left, root.val+right));
16
17 max[0] = Math.max(max[0],Math.max(ans,root.val+left+right));
18
19 return ans;
20
21 }