687.最长同值路径[DFS]题解

思路

求解最长同值路径，即求解树中两个节点之间路径的最大边数，且这个路径上的所有节点值是相同的。
那么针对一颗二叉树，这条路径：

1. 路径长度不为0，那么这条路径一定会经过某子树的根节点，因此该路径以该根节点为界，分为左右两段路径分别进行计算。
2. 路径长度为0，即为一个叶子节点。

考虑情况1，我们以任意一个子树的根节点r为例：

1. 若r.left.val==r.val 则我们需要知道其左子树的路径长度，并加1，得到当前左侧路径的值leftLength；若r.left.val!=r.val，leftLength直接赋值0即可。
2. 若r.right.val==r.val 则我们需要知道其右子树的路径长度，并加1，得到当前右侧路径的值rightLength；若r.right.val!=r.val，rightLength直接赋值0即可。
3. 通过两者相加计算得到当前子树中的路径长度curLength，并与已计算得到的最大路径长度进行比较。
4. 该子树计算结束，需要返回至该子树的父节点，需要注意的是这里返回的不是curLength，而是max(leftLength, rightLength)。因为经过其父节点的路径是不会同时经过该子树两侧的路径的，因此只需要取左右路径中的最大值。

因此我们需要利用到树的深度优先遍历，先计算子树中的路径，并从下往上返回。

代码如下：

public class Solution {
    private int maxLength;

    public int longestUnivaluePath(TreeNode root) {
        dfs(root);
        return maxLength;
    }

    //后续遍历，先处理左子树，再处理右子树，然后加上根节点进行一次计算并返回
    //不对
    private int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }

        int left, right;
        int leftLength = 0, rightLength = 0, curLength = 0;
        left = dfs(root.left);
        if (root.left != null && root.left.val == root.val) {
            leftLength = left + 1;
        }
        right = dfs(root.right);
        if (root.right != null && root.right.val == root.val) {
            rightLength = right + 1;
        }
        curLength = leftLength + rightLength;
        //经过当前根节点的最大路径长度
        if (curLength > maxLength) {
            maxLength = curLength;
        }
        return Math.max(leftLength, rightLength);
    }