Welcome to Subscribe On Youtube
Question
Formatted question description: https://leetcode.ca/all/314.html
Given the root
of a binary tree, return the vertical order traversal of its nodes' values. (i.e., from top to bottom, column by column).
If two nodes are in the same row and column, the order should be from left to right.
Example 1:
Input: root = [3,9,20,null,null,15,7] Output: [[9],[3,15],[20],[7]]
Example 2:
Input: root = [3,9,8,4,0,1,7] Output: [[4],[9],[3,0,1],[8],[7]]
Example 3:
Input: root = [3,9,8,4,0,1,7,null,null,null,2,5] Output: [[4],[9,5],[3,0,1],[8,2],[7]]
Constraints:
- The number of nodes in the tree is in the range
[0, 100]
. -100 <= Node.val <= 100
Algorithm
How to determine the order of the columns, you can give the root node a sequence number of 0, and then start the traversal of the layer sequence. For any left child node, the sequence number is reduced by 1, and the right child node sequence number is increased by 1, so that the node value of the same column can be changed by the sequence number Put together
Another way, is to use a TreeMap
to create a mapping between the sequence number and its corresponding node value.
- advantage of using TreeMap is that its
automatic sorting
function allows the column to be sorted from left to right. Since the sequence traversal needs to use the queue, the queue cannot only be stored at this time for nodes, it is necessary to store a pair consisting of a sequence number and a node, so that the sequence number can be operated every time it is taken out, and the nodes in the queue are also assigned their correct sequence numbers.
Code
-
// Time complexity O(n) // Space complexity O(n) public class Solution { private class TreeColumnNode{ public TreeNode treeNode; int col; public TreeColumnNode(TreeNode node, int col) { this.treeNode = node; this.col = col; } } public List<List<Integer>> verticalOrder(TreeNode root) { List<List<Integer>> res = new ArrayList<>(); if(root == null) { return res; } Queue<TreeColumnNode> queue = new LinkedList<>(); Map<Integer, List<Integer>> map = new HashMap<>(); queue.offer(new TreeColumnNode(root, 0)); int curLevelCount = 1; int nextLevelCount = 0; while(!queue.isEmpty()) { TreeColumnNode node = queue.poll(); if(map.containsKey(node.col)) { map.get(node.col).add(node.treeNode.val); } else { map.put(node.col, new ArrayList<Integer>(Arrays.asList(node.treeNode.val))); } curLevelCount--; if(node.treeNode.left != null) { queue.offer(new TreeColumnNode(node.treeNode.left, node.col - 1)); nextLevelCount++; } if(node.treeNode.right != null) { queue.offer(new TreeColumnNode(node.treeNode.right, node.col + 1)); nextLevelCount++; } if(curLevelCount == 0) { curLevelCount = nextLevelCount; nextLevelCount = 0; } } return new ArrayList<List<Integer>>(map.values()); } } // Time complexity O(nlogn) // Space complexity O(n) class Solution { public List<List<Integer>> verticalOrder(TreeNode root) { List<List<Integer>> res = new ArrayList<>(); if (root == null) return res; Queue<TreeNode> q = new LinkedList<>(); Queue<Integer> cols = new LinkedList<>(); Map<Integer, List<Integer>> map = new TreeMap<>(); q.offer(root); cols.offer(0); int min = 0, max = 0; while (!q.isEmpty()) { TreeNode cur = q.poll(); int col = cols.poll(); if (!map.containsKey(col)) map.put(col, new ArrayList<>()); map.get(col).add(cur.val); if (cur.left != null) { q.offer(cur.left); cols.offer(col - 1); min = Math.min(min, col - 1); } if (cur.right != null) { q.offer(cur.right); cols.offer(col + 1); max = Math.max(max, col + 1); } } res.addAll(map.values()); return res; } } ############ /** * 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>> verticalOrder(TreeNode root) { List<List<Integer>> ans = new ArrayList<>(); if (root == null) { return ans; } Deque<Pair<TreeNode, Integer>> q = new ArrayDeque<>(); q.offer(new Pair<>(root, 0)); TreeMap<Integer, List<Integer>> d = new TreeMap<>(); while (!q.isEmpty()) { for (int n = q.size(); n > 0; --n) { var p = q.pollFirst(); root = p.getKey(); int offset = p.getValue(); d.computeIfAbsent(offset, k -> new ArrayList()).add(root.val); if (root.left != null) { q.offer(new Pair<>(root.left, offset - 1)); } if (root.right != null) { q.offer(new Pair<>(root.right, offset + 1)); } } } return new ArrayList<>(d.values()); } }
-
// OJ: https://leetcode.com/problems/binary-tree-vertical-order-traversal/ // Time: O(NlogW) where N is the number of nodes in the tree and W is the width of the tree // Space: O(N) class Solution { public: vector<vector<int>> verticalOrder(TreeNode* root) { if (!root) return {}; map<int, vector<int>> m; queue<pair<TreeNode*, int>> q; q.emplace(root, 0); while (q.size()) { auto [node, index] = q.front(); q.pop(); m[index].push_back(node->val); if (node->left) q.emplace(node->left, index - 1); if (node->right) q.emplace(node->right, index + 1); } vector<vector<int>> ans; for (auto &[index, v] : m) ans.push_back(v); return ans; } };
-
# Definition for a binary tree node. # class TreeNode: # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right ''' >>> d = {1:'a', -1:'b', 10:'c', -100:'d'} >>> >>> d.items() {1: 'a', 10: 'c', -100: 'd', -1: 'b'} >>> ds = sorted(d.items()) >>> ds [(-100, 'd'), (-1, 'b'), (1, 'a'), (10, 'c')] >>> sorted(d.items(), key=lambda l: l[1]) [(1, 'a'), (-1, 'b'), (10, 'c'), (-100, 'd')] >>> dict(sorted(d.items(), key=lambda l: l[1])) {1: 'a', 10: 'c', -100: 'd', -1: 'b'} >>> [v for i,v in sorted(d.items(), key=lambda l: l[0])] ['d', 'b', 'a', 'c'] ''' class Solution: def verticalOrder(self, root: Optional[TreeNode]) -> List[List[int]]: if root is None: return [] q = deque([(root, 0)]) d = defaultdict(list) while q: for _ in range(len(q)): root, offset = q.popleft() d[offset].append(root.val) if root.left: q.append((root.left, offset - 1)) if root.right: q.append((root.right, offset + 1)) return [v for _, v in sorted(d.items())] ################# from collections import defaultdict class Solution(object): def verticalOrder(self, root): """ :type root: TreeNode :rtype: List[List[int]] """ def dfs(p, i, j, res): # i -> depth, j -> vertical-shift if p: res[j].append((p.val, i)) self.leftMost = min(j, self.leftMost) dfs(p.left, i + 1, j - 1, res) dfs(p.right, i + 1, j + 1, res) self.leftMost = float("inf") ans = [] res = defaultdict(list) dfs(root, 0, 0, res) i = self.leftMost while True: if not res[i]: break ans.append([item[0] for item in sorted(res[i], key=lambda a: a[1])]) i += 1 return ans
-
/** * Definition for a binary tree node. * type TreeNode struct { * Val int * Left *TreeNode * Right *TreeNode * } */ func verticalOrder(root *TreeNode) [][]int { ans := [][]int{} if root == nil { return ans } d := map[int][]int{} q := []pair{pair{root, 0} } for len(q) > 0 { for n := len(q); n > 0; n-- { p := q[0] q = q[1:] root = p.node offset := p.offset d[offset] = append(d[offset], root.Val) if root.Left != nil { q = append(q, pair{root.Left, offset - 1}) } if root.Right != nil { q = append(q, pair{root.Right, offset + 1}) } } } idx := []int{} for i := range d { idx = append(idx, i) } sort.Ints(idx) for _, i := range idx { ans = append(ans, d[i]) } return ans } type pair struct { node *TreeNode offset int }