[编程题]树上最长单色路径

题目描述:
对于一棵由黑白点组成的二叉树,我们需要找到其中最长的单色简单路径,其中简单路径的定义是从树上的某点开始沿树边走不重复的点到树上的另一点结束而形成的路径,而路径的长度就是经过的点的数量(包括起点和终点)。而这里我们所说的单色路径自然就是只经过一种颜色的点的路径。你需要找到这棵树上最长的单色路径。
给定一棵二叉树的根节点(树的点数小于等于300,请做到O(n)的复杂度),请返回最长单色路径的长度。这里的节点颜色由点上的权值表示,权值为1的是黑点,为0的是白点。

public class LongestPurePath {

    /**
     * @param args
     */
    int maxLen = 0;
    public int findPath(TreeNode root) {
        // write code here
        if(root == null){
            return 0;
        }
        getPath(root);
        return maxLen;
    }
    //获取左右子树中最长的路径值
    public int getPath(TreeNode root) {
        if(root == null){
            return 0;
        }
        //记录左右子数的最大路径长度
        int left = 0;
        int right = 0;
        if(root.left!=null){
            int len = getPath(root.left); //左子树的最大长度
            if(root.val == root.left.val){ //颜色相同
                left = len;
            }
        }
        if(root.right!=null){
            int len = getPath(root.right); //左子树的最大长度
            if(root.val == root.right.val){ //颜色相同
                right = len;
            }
        }
        maxLen = Math.max(maxLen, left+right+1); //记录最大长度值,包含左子树最大路径+根节点+右子树
        return Math.max(left, right) + 1; //返回左右子树中最长的路径值
    }
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值