leetcode 687 最长同值路径

给定一个二叉树的 root ,返回 最长的路径的长度 ,这个路径中的 每个节点具有相同值。 这条路径可以经过也可以不经过根节点。

做这道题的时候,我一开始想到的是直接从根节点往下遍历,然后用哈希表记录相同节点出现的次数,然后取出出现次数最多的元素
代码类似这样

    public void longestUnivaluePath(TreeNode node, Map<Integer, Integer> map) {
        if (node == null) {
            return;
        }
        //节点相同则记录次数
        if (node.left != null && node.val == node.left.val) {
            map.put(node.val,
                    map.getOrDefault(node.val, 0) + 1);
        }

        if (node.right != null && node.val == node.right.val) {
            map.put(node.val,
                    map.getOrDefault(node.val, 0) + 1);
        }
        longestUnivaluePath(node.left, map);
        longestUnivaluePath(node.right, map);
    }

然后发现这样子虽然能够判断相同元素出现最多的次数,但是无法判断他们是否是相连的

比如这种情况

后来看题解,是将他看成是解决以某个节点为根节点的左子树和右子树的最长同值路径

   public int longestUnivaluePathV2(TreeNode root) {
        if (root == null) {
            return 0;
        }
        //当前根节点左子树最大的路径
        int left = longestUnivaluePathV2(root.left);
        //当前根节点右子树最大的路径
        int right = longestUnivaluePathV2(root.right);

        //记录能否能和当前节点连接 
        int l = 0, r = 0;

        //如果能和父节点连起来 ,长度 + 1 , 没有则返回0
        if (root.left != null && root.left.val == root.val) {
            l = left + 1;
        }

        if (root.right != null && root.right.val == root.val) {
            r = right + 1;
        }

        //记录最大长度
        total = Math.max(total, l + r);

        //返回最大的路径长度,为什么只返回一边?因为路径只有一个起始一个末端
        //将这个子最长边继续
        return Math.max(l, r);
    }
posted @ 2022-09-06 15:54  阿弱  阅读(29)  评论(0编辑  收藏  举报