687. 最长同值路径

深度优先搜索

class Solution {

    int maxLength = 0;

    public int longestUnivaluePath(TreeNode root) {

        depth(root);
        return maxLength;
    }

    /**
     * 在《104. 二叉树的最大深度》的基础上
     * 顺便计算以root为起始节点的最长同值路径
     */
    public int depth(TreeNode root){

        if (root == null){
            return 0;
        }

        int maxSon = 0;

        int left = depth(root.left);
        int right = depth(root.right);

        /**
         * 上述是不包含根节点，即只包含左子树或者右子树的情况
         * 如果根节点的值和左右孩子都相等，即根节点在路径中间，还要单独判断
         */
        if (root.left != null && root.left.val == root.val && root.right != null && root.right.val == root.val){
            maxLength = Math.max(maxLength, left + right + 2);
        }

        /**
         * maxSon记录的只是depth()方法需要的最大子树的深度，所以只能是left或者right，这个结果对longestUnivaluePath()方法无用
         * 而maxLength记录的是左右子树加起来最长的路径，是left + right，也是主方法需要的结果
         */
        if (root.left != null && root.left.val == root.val){
            maxSon = left + 1;
        }

        if (root.right != null && root.right.val == root.val){
            maxSon = Math.max(maxSon, right + 1);
        }

        /**
         * 如果没有经过第一个if判断，需要在此记录下最大的子树深度
         */
        maxLength = Math.max(maxSon, maxLength);

        return maxSon;
    }
}

/**
 * 时间复杂度 O(n)
 * 空间复杂度 O(logn)
 */

https://leetcode-cn.com/problems/longest-univalue-path/

标签：right,val,路径,maxSon,maxLength,同值,root,687,left
来源： https://www.cnblogs.com/taoyuann/p/15743891.html