二叉树的垂序遍历(TreeMap)

给你二叉树的根结点 root ,请你设计算法计算二叉树的 垂序遍历 序列。

对位于 (row, col) 的每个结点而言,其左右子结点分别位于 (row + 1, col - 1) 和 (row + 1, col + 1) 。树的根结点位于 (0, 0) 。

二叉树的 垂序遍历 从最左边的列开始直到最右边的列结束,按列索引每一列上的所有结点,形成一个按出现位置从上到下排序的有序列表。如果同行同列上有多个结点,则按结点的值从小到大进行排序。

返回二叉树的 垂序遍历 序列。

示例 1:
在这里插入图片描述

输入:root = [3,9,20,null,null,15,7]
输出:[[9],[3,15],[20],[7]]
解释:
列 -1 :只有结点 9 在此列中。
列 0 :只有结点 3 和 15 在此列中,按从上到下顺序。
列 1 :只有结点 20 在此列中。
列 2 :只有结点 7 在此列中。
示例 2:

在这里插入图片描述

输入:root = [1,2,3,4,5,6,7]
输出:[[4],[2],[1,5,6],[3],[7]]
解释:
列 -2 :只有结点 4 在此列中。
列 -1 :只有结点 2 在此列中。
列 0 :结点 1 、5 和 6 都在此列中。
1 在上面,所以它出现在前面。
5 和 6 位置都是 (2, 0) ,所以按值从小到大排序,5 在 6 的前面。
列 1 :只有结点 3 在此列中。
列 2 :只有结点 7 在此列中。
示例 3:

在这里插入图片描述

输入:root = [1,2,3,4,6,5,7]
输出:[[4],[2],[1,5,6],[3],[7]]
解释:
这个示例实际上与示例 2 完全相同,只是结点 5 和 6 在树中的位置发生了交换。
因为 5 和 6 的位置仍然相同,所以答案保持不变,仍然按值从小到大排序。

提示:

树中结点数目总数在范围 [1, 1000] 内
0 <= Node.val <= 1000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:用

TreeMap<Integer,TreeMap<Integer,List< Integer >>>来排序,因为TreeMap本身就是按key来递增排序的。

最外层的Integer是col,再里一层的Integer是row,因为排序方式是从最小列到最大列,最小行到最大行。

最后遍历TreeMap,把同一列的 List< Integer >里的东西弄出来。

class Solution {
    TreeMap<Integer,TreeMap<Integer,List<Integer>>> treeMap = new TreeMap<>();

    public List<List<Integer>> verticalTraversal(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        put(root, 0, 0);
        Iterator iterator = treeMap.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry entry = (Map.Entry) iterator.next();
            TreeMap<Integer,List<Integer>> t = (TreeMap<Integer,List<Integer>>)entry.getValue();
            List<Integer> list = new ArrayList<>();
            Iterator iter = t.entrySet().iterator();
            while (iter.hasNext()) {
                Map.Entry ent = (Map.Entry) iter.next();
                List<Integer> l = (List<Integer>)ent.getValue();
                list.addAll(l);
            }
            result.add(list);
        }
        return result;
    }

    public void put(TreeNode root, int col, int row) {
        if (root == null) {
            return;
        }
        TreeMap<Integer,List<Integer>> t;
        List<Integer> list;
        if (treeMap.containsKey(col)) {
            t = treeMap.get(col);
            if (t.containsKey(row)) {
                list = t.get(row);
            }else {
                list = new ArrayList<Integer>();
            }
        }else {
            t = new TreeMap<>();
            list = new ArrayList<Integer>();
        }
        list.add(root.val);
        Collections.sort(list);
        t.put(row, list);
        treeMap.put(col, t);
        put(root.left, col-1, row+1);
        put(root.right, col+1, row+1);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值