题解 | #求二叉树的层序遍历#
求二叉树的层序遍历
https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ #include <vector> class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @return int整型vector<vector<>> */ vector<vector<int> > levelOrder(TreeNode* root) { if(!root) return{};//空树 vector<vector<int>> ans; ans.push_back({root->val}); queue<pair<TreeNode*,int>> record; if(root->left) record.push({root->left,1}); if(root->right) record.push({root->right,1}); while(!record.empty()){ int a=record.front().second;//记录层数 TreeNode* t=record.front().first;//记录节点 record.pop();//用完后删除 if(t->left) record.push({t->left,a+1});//存在左子树放入 if(t->right) record.push({t->right,a+1});//存在右子树放入 if(a==ans.size()) ans.push_back({t->val});//进入新的一层加入 else ans[a].push_back(t->val);//本层加入 } return ans; } };