文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 DFS
    • 2.2 BFS

1. 題目

給定一個二叉樹，返回其結點 垂直方向（從上到下，逐列）遍歷的值。

如果兩個結點在同一行和列，那么順序則為 從左到右。

示例 1： 輸入: [3,9,20,null,null,15,7]3/\/ \ 9 20/\/ \15 7 輸出: [[9],[3,15],[20],[7] ]示例 2: 輸入: [3,9,8,4,0,1,7]3/\/ \9 8/\ /\/ \ / \ 4 0 1 7 輸出: [[4],[9],[3,0,1],[8],[7] ]示例 3: 輸入: [3,9,8,4,0,1,7,null,null,null,2,5] （注意：0 的右側子節點為 2，1 的左側子節點為 5）3/\/ \9 8/\ /\/ \/ \4 01 7/\/ \5 2輸出: [[4],[9,5],[3,0,1],[8,2],[7] ]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/binary-tree-vertical-order-traversal
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

2.1 DFS

• 記錄深度、橫坐標，按橫坐標存入map，取出來的時候按深度排序

class Solution {map<int, vector<pair<int,int>>> ans; public:vector<vector<int>> verticalOrder(TreeNode* root) {dfs(root, 0, 0);vector<vector<int>> res;for(auto& a : ans){ //按橫坐標取出sort(a.second.begin(), a.second.end(),[&](auto a, auto b){return a.first < b.first;//按深度排序});vector<int> temp;for(auto& d_val : a.second)temp.push_back(d_val.second);res.push_back(temp);}return res;}void dfs(TreeNode* root, int x, int deep){if(!root) return;ans[x].push_back({deep, root->val});//深度、valdfs(root->left, x-1, deep+1);dfs(root->right, x+1, deep+1);} };

8 ms 8.7 MB

2.2 BFS

• 只需記錄x橫坐標，存入map，從頂向下BFS層次遍歷（保證從上到下的順序）

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

8 ms 8.1 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 314. 二叉树的垂直遍历（BFS/DFS）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。