給定二叉樹,按垂序遍歷返回其結點值。
對位于?(X, Y)?的每個結點而言,其左右子結點分別位于?(X-1, Y-1)?和?(X+1, Y-1)。
把一條垂線從?X = -infinity?移動到?X = +infinity?,每當該垂線與結點接觸時,我們按從上到下的順序報告結點的值( Y?坐標遞減)。
如果兩個結點位置相同,則首先報告的結點值較小。
按?X?坐標順序返回非空報告的列表。每個報告都有一個結點值列表。
?
示例 1:
輸入:[3,9,20,null,null,15,7]
輸出:[[9],[3,15],[20],[7]]
解釋:?
在不喪失其普遍性的情況下,我們可以假設根結點位于 (0, 0):
然后,值為 9 的結點出現在 (-1, -1);
值為 3 和 15 的兩個結點分別出現在 (0, 0) 和 (0, -2);
值為 20 的結點出現在 (1, -1);
值為 7 的結點出現在 (2, -2)。
示例 2:輸入:[1,2,3,4,5,6,7]
輸出:[[4],[2],[1,5,6],[3],[7]]
解釋:
根據給定的方案,值為 5 和 6 的兩個結點出現在同一位置。
然而,在報告 "[1,5,6]" 中,結點值 5 排在前面,因為 5 小于 6。
?提示:
樹的結點數介于 1?和?1000?之間。
每個結點值介于?0?和?1000?之間。來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
解法:
?
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/
class Solution {
public:vector<vector<int>> verticalTraversal(TreeNode* root) {map<int, vector<int> > m; queue<pair<int, TreeNode*> > q; q.push(make_pair(0, root)); while (!q.empty()){set<pair<int, int> > tmp; int len = q.size();for (int i = 0; i < len; ++i){auto p = q.front(); q.pop();tmp.insert(make_pair(p.first, p.second->val));if (p.second->left) q.push(make_pair(p.first - 1, p.second->left));if (p.second->right) q.push(make_pair(p.first + 1, p.second->right));}for (auto p : tmp) m[p.first].push_back(p.second);}vector<vector<int> > res;for (auto kv : m) res.push_back(kv.second);return res;}
};
?