树上最长单色路径

来源:互联网 发布:软件授权码生成器 编辑:程序博客网 时间:2024/04/27 03:42

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

给定一棵二叉树的根节点(树的点数小于等于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;     }}

后序遍历的思想,同时记录当前root能达到的最大深度(颜色相同),另外加一个字段记录根节点可以到达的最大深度

这题目其实挺麻烦的,写得我都崩溃了。。。。

0 0
原创粉丝点击