Given a binary tree, find the length of the longest path where each node in the path has the same value. This path may or may not pass through the root.
Note: The length of path between two nodes is represented by the number of edges between them.
Example 1:
Input:
5 / \ 4 5 / \ \ 1 1 5
Output:
2
Example 2:
Input:
1 / \ 4 5 / \ \ 4 4 5
Output:
2
Note: The given binary tree has not more than 10000 nodes. The height of the tree is not more than 1000.
后序遍历,程序如下所示:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private int len = 0;
public int longestUnivaluePath(TreeNode root) {
if (root == null){
return 0;
}
traversal(root, root.val);
return len;
}
public int traversal(TreeNode root, int pre){
if (root == null){
return 0;
}
int left = traversal(root.left, root.val);
int right = traversal(root.right, root.val);
len = Math.max(len, left + right);
if (root.val == pre){
return Math.max(left, right) + 1;
}
return 0;
}
}