【BFS】LeetCode 314. 二叉树的垂直遍历

题目链接

314. 二叉树的垂直遍历

思路

如图所示，可以对输出列次顺序作如下标记。

将每一列的列号作为 key，同时将该列的所有结点组成的 List 作为 value，使用 HashMap<Integer, ArrayList<Integer>> 进行存储。

注意： 在进行 BFS 的过程中要存储最小层数下标，只有这样才知道 BFS 结束之后从哪个层开始遍历结果。

代码

class Solution {
    public List<List<Integer>> verticalOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if(root == null){
            return result;
        }

        int minIndex = 0;
        HashMap<Integer, ArrayList<Integer>> map = new HashMap<>();
        Queue<TreeNode> nodeQueue = new LinkedList<>();
        Queue<Integer> indexQueue = new LinkedList<>();

        nodeQueue.offer(root);
        indexQueue.offer(0);
        while(!nodeQueue.isEmpty()){
            int cnt = nodeQueue.size();
            while(cnt > 0){
                cnt--;
                TreeNode node = nodeQueue.poll();
                int currentIndex = indexQueue.poll();
                if(!map.containsKey(currentIndex)){
                    map.put(currentIndex, new ArrayList<>());
                }

                ArrayList<Integer> temp = map.get(currentIndex);
                temp.add(node.val);
                if(node.left != null){
                    nodeQueue.add(node.left);
                    indexQueue.add(currentIndex - 1);
                    minIndex = Math.min(minIndex, currentIndex - 1);
                }
                if(node.right != null){
                    nodeQueue.add(node.right);
                    indexQueue.add(currentIndex + 1);
                }
            }
        }
        
        for(int i = minIndex; i < minIndex + map.size(); i++){
            result.add(map.get(i));
        }

        return result;
    }
}