Skip to main content

Command Palette

Search for a command to run...

987. Vertical Order Traversal of a Binary Tree

Published
3 min read

https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree/

problems

Given the root of a binary tree, calculate the vertical order traversal of the binary tree.

For each node at position (row, col), its left and right children will be at positions (row + 1, col - 1) and (row + 1, col + 1) respectively. The root of the tree is at (0, 0).

The vertical order traversal of a binary tree is a list of top-to-bottom orderings for each column index starting from the leftmost column and ending on the rightmost column. There may be multiple nodes in the same row and same column. In such a case, sort these nodes by their values.

Return the vertical order traversal of the binary tree.

example

Example 1:

image.png

Input: root = [3,9,20,null,null,15,7]
Output: [[9],[3,15],[20],[7]]
Explanation:
Column -1: Only node 9 is in this column.
Column 0: Nodes 3 and 15 are in this column in that order from top to bottom.
Column 1: Only node 20 is in this column.
Column 2: Only node 7 is in this column.

Example 2:

image.png

Input: root = [1,2,3,4,5,6,7]
Output: [[4],[2],[1,5,6],[3],[7]]
Explanation:
Column -2: Only node 4 is in this column.
Column -1: Only node 2 is in this column.
Column 0: Nodes 1, 5, and 6 are in this column.
          1 is at the top, so it comes first.
          5 and 6 are at the same position (2, 0), so we order them by their value, 5 before 6.
Column 1: Only node 3 is in this column.
Column 2: Only node 7 is in this column.

Example 3:

image.png

Input: root = [1,2,3,4,6,5,7]
Output: [[4],[2],[1,5,6],[3],[7]]
Explanation:
This case is the exact same as example 2, but with nodes 5 and 6 swapped.
Note that the solution remains the same since 5 and 6 are in the same location and should be ordered by their values.

Constraints:

The number of nodes in the tree is in the range [1, 1000].
0 <= Node.val <= 1000

solution

思路:

  1. 把所有节点的坐标遍历记录下来。
  2. 按照纵坐标顺序,行坐标从小到大
  3. 细节,同样的坐标,值从小到到。 可以用嵌套的TreeMap 纵坐标为外层 key,横坐标为内层坐标key,这样遍历坐标符合从小到大。由于坐标可能重叠,用List 保存值,读取的时候需要注意顺序。
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> verticalTraversal(TreeNode root) {
        TreeMap<Integer, TreeMap<Integer, List<Integer>>> map = new TreeMap<>();
        dfs(root,0, 0, map);
        //
        List<List<Integer>> ret = new ArrayList<>();
        for (Map.Entry<Integer, TreeMap<Integer, List<Integer>>> entry: map.entrySet()) {
            List<Integer> tmp = new ArrayList<>();
            for (Map.Entry<Integer, List<Integer>> entry1: entry.getValue().entrySet()) {
                List<Integer> list = entry1.getValue();
                Collections.sort(list);
                tmp.addAll(list);
            }
            if (tmp.size() >0) {
                ret.add(tmp);
            }
        }
        return ret;
    }
    public void dfs(TreeNode n, int x, int y, TreeMap<Integer, TreeMap<Integer, List<Integer>>> m) {
        if (n == null) {
            return;
        }
        //vist n;
        m.putIfAbsent(y, new TreeMap<>());
        TreeMap<Integer, List<Integer>> sub = m.get(y);
        sub.putIfAbsent(x, new ArrayList<Integer>());
        m.get(y).get(x).add(n.val);
        // vist left;
        dfs(n.left, x+1, y-1, m);
        // vist right;
        dfs(n.right, x+1, y+1, m);
    }
}

More from this blog

Google Best Practice For Java Libraries (JLBP-0)

this is a chinese tranlation from Google Best Practice For Java Libraries Google Java库最佳实践 Google Java库的最佳实践是一些规则,旨在减少使用相关的Java库时的问题。这些实践来自于多年来维护开源Java库的经验,并且受到了从错误中吸取的许多宝贵教训的启发。我们发现遵循这些规则可以得到质量更高的Java库,减少依赖冲突和其他类型的问题。这个规则列表是开放的,所以可能会不时地添加新的规则。 最佳实践...

Nov 22, 20231 min read

【阅读】uml精粹 类图

本书的整体脉络如下: class diagram 表示对象的类型以及其间存在的静态关系。用一个长方形表示类,用连线表示类之间的关系。 具体类图 长方形 对象的类型,也就是类有三部分表示而成 类名 (class name) 属性 (attribute) 操作 (operation) 属性 (attribute) 其中抽象来说,类的成员变量 可以有两种表示, 属性表示 和 关联表示。 属性表示:就是画在框里, 一般写 包括 可见行, 类型,名称 ,也可以写默认值 和 注释。 也可以标注是否是 ...

Sep 4, 20221 min read

【夜读源码】grpc-java 客户端发送 (一)

快速回顾 接上文,一次 grpc 的请求包括四个步骤,客户端发送请求到网络,是第一步。发送过程,Stub 把请求传给了ClientCall, ClientCall 传给了 ClientStream,ClientStream 有Netty 的实现,内部会把数据传输到网络上。 本文主要介绍 blockingUnaryCall 是如何使用 ClientCall ClientChannel 工作的, waitAndDrain 和 ClientCall 的方法如何对应。 Stub 外层简述 最外层的Stu...

Sep 4, 20224 min read
【夜读源码】grpc-java  客户端发送 (一)

【夜读源码】grpc-java 概览

gRPC 是 Google 开源的一套语言无关基于HTTP2协议的 RPC 框架,其中 Java 的实现称为 grpc-java,代码地址在 https://github.com/grpc/grpc-java。 RPC (remote process call)的含义是程序执行本地的一个方法(看上去像本地执行),实际上会在远程程序代码内执行,这个过程会涉及到网络通讯和传输协议。gRPC作为 RPC 一种实现,传输的网络协议是HTTP2,传输的内容按照 Protocol Buffer 进行编码,并...

Aug 16, 20223 min read
【夜读源码】grpc-java 概览

喝一杯mockito

mockito 是最常用的 java mock 库, 用于模拟创建各种对象和行为。本文会介绍一下 基础库 mockitio的用法,进行一些练习。mockitio 通常不单独使用,配合 Junit一起使用。 下面列举一下用法, 模拟对象 创建mock 对象,并且可以要求,这个对象的成员方法符合我们的行为要求。 Foo foo= Mockito.mock(Foo.class); 直接返回 直接返回,我们需要的值 when thenReturn @Test void returnWh...

Jul 20, 20222 min read
喝一杯mockito

Xiao Kun's Infra Insight Blog

14 posts

Focus on interpreting FANG/Big Factory Engineering Blog, extracting practical experience of Infra that can be implemented