LeetCode 314. 二叉树的垂直遍历（median）

题目

LeetCode 314. 二叉树的垂直遍历（median）

给你一个二叉树的根结点，返回其结点按 垂直方向（从上到下，逐列）遍历的结果。

如果两个结点在同一行和列，那么顺序则为 从左到右

示例一：

输入：root = [3,9,20,null,null,15,7]
输出：[[9],[3,15],[20],[7]]

示例二：

输入：root = [3,9,8,4,0,1,7]
输出：[[4],[9],[3,0,1],[8],[7]]

示例三：

输入：root = [3,9,8,4,0,1,7,null,null,null,2,5]
输出：[[4],[9,5],[3,0,1],[8,2],[7]]

提示：

• 树中结点的数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

思路

题目中的垂直位置可以按如下方式表示：
（红色代表位置的变化，蓝色代表结点投影下来的位置）

1. 设根节点的水平位置为0
2. 从根节点出发，向左则位置减一，向右则位置加一

使用vertical保存每一水平位置的结点的val。由于层次遍历就是按照从上到下、从左到右的，所以每一水平位置对应的vector中的结点的值就是按照题目要求排列的

代码

C++实现：

class Solution {
public:
    vector<vector<int>> verticalOrder(TreeNode* root) {
        if (!root)
            return {};
        map<int, vector<int>> vertical;
        queue<pair<TreeNode*, int>> q;
        q.push(make_pair(root, 0));
        while (!q.empty()) {
            int sz = q.size();
            while (sz--) {
                auto p = q.front();
                q.pop();
                vertical[p.second].push_back(p.first->val);
                if (p.first->left)
                    q.push(make_pair(p.first->left, p.second - 1));
                if (p.first->right)
                    q.push(make_pair(p.first->right, p.second + 1));
            }
        }
        
        int idx = 0;
        vector<vector<int>> ret(vertical.size());
        for (auto& p : vertical) {
            for (auto& rt : p.second) {
                ret[idx].push_back(rt);
            }
            ++idx;
        }
        return ret;
    }
};

Java实现：感谢Lenaxiao提供

class Solution {
    public List<List<Integer>> verticalOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        Map<Integer, List<Integer>> map = new TreeMap<>();
        
        Queue<TreeNode> q = new LinkedList<>();
        Queue<Integer> ids = new LinkedList<>();
        q.add(root);
        ids.add(0);
        while (!q.isEmpty()) {
            int size = q.size();
            for (int i = 0; i < size; i++) {
                TreeNode cur = q.poll();
                int id = ids.poll();

                if (!map.containsKey(id)) {
                    map.put(id, new ArrayList<>());
                }
                map.get(id).add(cur.val);

                if (cur.left != null) {
                    q.add(cur.left);
                    ids.add(id - 1);
                }
                if (cur.right != null) {
                    q.add(cur.right);
                    ids.add(id + 1);
                }
            }
        }
        return new ArrayList<>(map.values());
    }
}

---

有收获的话，求个赞~ Click Here

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/binary-tree-vertical-order-traversal/