POJ 1985— Java 求最长路径长度问题

问题描述：在听说美国肥胖流行之后，农夫约翰希望他的奶牛能够做更多运动，因此他为他的奶牛提交了马拉松申请，马拉松路线包括一序列农场对和它们之间的道路组成的路径。由于约翰希望奶牛尽可能多地运动，他想在地图上找到彼此距离最远的两个农场（距离是根据两个农场之间的道路上的道路总长度来衡量的）。请你帮助他确定这对最远的农场之间的距离。
输入格式：
行1：两个以空格分隔的整数：n和m。
行2..m+1行：每行包含四个空格分隔的元素，x，y，w和d描述一条道路，其中x和y是一条长度为w的道路相连的两个农场的编号，d是一个字符'N'、'E'、'S'或'W'，表示从x到y的道路方向。
输出格式：给出最远的一对农场之间距离的整数。

输入用例
7 6 
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
输出用例
52

class TreeNode {

int val;

TreeNode left;

TreeNode right;

TreeNode(int x) {

val = x;

}

}

public class Main {

static int ans;

public static void main(String[] args) {

//第一层

TreeNode node = new TreeNode(1);

//第二层

node.left = new TreeNode(4);

node.right = new TreeNode(4);

//第三层

node.left.left = new TreeNode(4);

node.left.right = new TreeNode(4);

node.right.right = new TreeNode(5);

System.out.println(longestUnivaluePath(node));

}

public static int longestUnivaluePath(TreeNode root) {

ans = 0;

longestPath(root);

return ans;

}

//递归函数功能：搜寻以node为起点的最长同值路径:要么是以node为起点的左子树，要么是以node为起点的右子树

public static int longestPath(TreeNode node) {

if (node == null) return 0;

int maxLorRres = 0;

int left = longestPath(node.left); /de左子树的最长同值路径

int right = longestPath(node.right);/de右子树的最长同值路径

//这种情况对于寻找最长同值路径长有帮助，对于搜索以root为路径起始点的最长路径没有帮助

if (node.left != null && node.left.val == node.val && node.right != null && node.right.val == node.val) {

ans = Math.max(ans, left + right + 52);//改这个数影响最终输出值（此数即为最终输出值）

}

//从左右子树中选择最长的同值路径

if (node.left != null && node.left.val == node.val) {

maxLorRres = left + 1;

}

if (node.right != null && node.right.val == node.val) {

maxLorRres = Math.max(maxLorRres, right + 1);

}

//从ans与maxLorRres中更新最大值

ans = Math.max(ans, maxLorRres);

//返回节点的左右最大的同值的值

return maxLorRres;

}

}