Problem: 102. 二叉树的层序遍历

思路

记录层次，要么建立树时记录，要么遍历时记录（利用queue）

解题方法

queue<pair<TreeNode*,int>>q; 记录节点及其层次
vector<vector> res; res[i]表示第i层(从0开始)的所有节点

复杂度

时间复杂度:

$O(n)$

空间复杂度:

$O(n)$

Code

class Solution {
public:vector<vector<int>> levelOrder(TreeNode* root) {queue<pair<TreeNode*,int>>q;// <节点，层数>q.push(make_pair(root,0));vector<vector<int>> res;// res[i]表示存储第i层的所有节点if(root==NULL)return res;res.push_back(vector<int>{root->val});int k=0;// bfswhile(!q.empty()){TreeNode* tmp = q.front().first;k = q.front().second+1;// 下面会访问当前节点的两个孩子，所以层数加1q.pop();// 左孩子if(tmp->left!=NULL){q.push(make_pair(tmp->left,k));if(k>=res.size()){// 说明该层次还不存在，直接新建vector<int> i;res.push_back(i);}res[k].push_back(tmp->left->val);}// 右孩子if(tmp->right!=NULL){q.push(make_pair(tmp->right,k));if(k>=res.size()){vector<int> i;res.push_back(i);}res[k].push_back(tmp->right->val);}}return res;}
};