树上最长单色路径

最新推荐文章于 2025-03-31 22:48:33 发布

ForABiggerWorld

最新推荐文章于 2025-03-31 22:48:33 发布

阅读量569

点赞数

CC 4.0 BY-SA版权

文章标签：二叉树

本文链接：https://blog.youkuaiyun.com/zjucor/article/details/52643340

在一颗由黑白点组成的二叉树中，求解最长的单色简单路径，即路径只经过同色节点且不重复。要求路径长度。采用后序遍历策略，结合节点颜色信息，动态维护以当前节点为根的最大同色深度。题目复杂度要求为O(n)，实现过程中存在挑战。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

对于一棵由黑白点组成的二叉树，我们需要找到其中最长的单色简单路径，其中简单路径的定义是从树上的某点开始沿树边走不重复的点到树上的另一点结束而形成的路径，而路径的长度就是经过的点的数量(包括起点和终点)。而这里我们所说的单色路径自然就是只经过一种颜色的点的路径。你需要找到这棵树上最长的单色路径。

给定一棵二叉树的根节点(树的点数小于等于300，请做到O(n)的复杂度)，请返回最长单色路径的长度。这里的节点颜色由点上的权值表示，权值为1的是黑点，为0的是白点。

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/

// 后序遍历的思想

public class LongestPath {

    private int max = 1;

    public int findPath(TreeNode root) {

        Traversal(root);

        //if(root == null) return 0;

        return max ;
    }

    // 从当前节点能到达的最大深度
    public int Traversal(TreeNode root) {

        if(root == null) return 0;

        int left  = Traversal(root.left);
        int right = Traversal(root.right);
        int len = 1;

        if(root.right == null && root.left == null) return len;

        if(root.right != null && root.right.val == root.val)
            len = right + 1;
        if(root.left != null && root.left.val == root.val)
            len = Math.max(len, left + 1);

        if(root.right == null || root.left == null)
            max = Math.max(max, len);
        else
            if(root.right.val == root.val && root.val == root.left.val)
                max = Math.max(max, left + right + 1);
            else
                max = Math.max(max, len);

        return len; 
    }
}