LeetCode 987. 二叉树的垂序遍历(递归/循环)
生活随笔
收集整理的這篇文章主要介紹了
LeetCode 987. 二叉树的垂序遍历(递归/循环)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
1. 題目
給定二叉樹,按垂序遍歷返回其結點值。
對位于 (X, Y) 的每個結點而言,其左右子結點分別位于 (X-1, Y-1) 和 (X+1, Y-1)。
把一條垂線從 X = -infinity 移動到 X = +infinity ,每當該垂線與結點接觸時,我們按從上到下的順序報告結點的值( Y 坐標遞減)。
如果兩個結點位置相同,則首先報告的結點值較小。
按 X 坐標順序返回非空報告的列表。每個報告都有一個結點值列表。
示例 1:
示例 2:
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
2. 解題
- map的key記錄x坐標,value記錄點的集合{val, 深度}
- 對x一樣的點集,按深度第一,值第二進行排序
2.1 遞歸
class Solution { map<int, vector<vector<int>>> m;//x坐標,節點集合< <val, deep> > public:vector<vector<int>> verticalTraversal(TreeNode* root) {//前序遍歷 根左右if(!root)return {};dfs(root,0,0);vector<vector<int>> temp;vector<vector<int>> ans(m.size());int i = 0, j;for(auto it = m.begin(); it != m.end(); ++it){temp = it->second;//點集合sort(temp.begin(), temp.end(),[&](auto a, auto b){if(a[1] == b[1])return a[0] < b[0];//深度一樣,按值return a[1] < b[1];//深度小的在前});for(j = 0; j < temp.size(); ++j)ans[i].push_back(temp[j][0]);//數值寫入答案i++;}return ans;}void dfs(TreeNode* root, int x, int deep){if(!root)return;m[x].push_back({root->val,deep});dfs(root->left, x-1, deep+1);dfs(root->right, x+1, deep+1);} };24 ms 16.3 MB
2.2 層序遍歷
class Solution { public:vector<vector<int>> verticalTraversal(TreeNode* root) {map<int, vector<vector<int>>> m;//x坐標,節點集合< <val, deep> >if(!root)return {};queue<pair<TreeNode*,pair<int,int>>> q;//節點及其坐標x,yq.push({root,{0,0}});pair<TreeNode*,pair<int,int>> tp;TreeNode* node;int x, y;while(!q.empty()){tp = q.front();q.pop();node = tp.first;x = tp.second.first;y = tp.second.second;m[x].push_back(vector<int> {node->val,y});if(node->left)q.push({node->left, {x-1,y+1}});if(node->right)q.push({node->right, {x+1,y+1}});}vector<vector<int>> temp;vector<vector<int>> ans(m.size());int i = 0, j;for(auto it = m.begin(); it != m.end(); ++it){temp = it->second;sort(temp.begin(), temp.end(),[&](auto a, auto b){if(a[1] == b[1])return a[0] < b[0];//深度一樣,按值return a[1] < b[1];//深度小的在前});for(j = 0; j < temp.size(); ++j)ans[i].push_back(temp[j][0]);//數值寫入答案i++;}return ans;} };16 ms 13.2 MB
創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的LeetCode 987. 二叉树的垂序遍历(递归/循环)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode 971. 翻转二叉树以
- 下一篇: LeetCode MySQL 1075.