题目
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
提示:
- 树中节点数目在范围 [0, 2000] 内
- -1000 <= Node.val <= 1000
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> v;
if(!root) return v;
int curLevel = 1,NodeCount = 0;
// 保存每个节点对应的层数
map<TreeNode*,int> m;
queue<TreeNode*> q;
q.push(root);
m.insert(pair<TreeNode*,int>(root,1));
vector<int> temp;
while(!q.empty()){
root = q.front();
q.pop();
// 如果目前的层数等于节点的层数,说明节点还在这个一层
if(curLevel==m[root]) temp.push_back(root->val);
// 说明节点不在这一层
else{
v.push_back(temp);
temp.clear();
temp.push_back(root->val);
curLevel++;
}
if(root->left){
q.push(root->left);
m.insert(pair<TreeNode*,int>(root->left,curLevel+1));
}
if(root->right){
q.push(root->right);
m.insert(pair<TreeNode*,int>(root->right,curLevel+1));
}
}
// 最后一层要特殊添加
v.push_back(temp);
return v;
}
};