题解 | #求二叉树的层序遍历#
求二叉树的层序遍历
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 <queue> class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @return int整型vector<vector<>> */ // 借助队列处理,先进入父节点,在出队的时候放入其子节点 vector<vector<int> > levelOrder(TreeNode* root) { // write code here vector< vector<int> > a_array; if (root == nullptr) { return a_array; } queue<TreeNode *> queue; queue.push(root); while (!queue.empty()) { vector<int> array; int size = queue.size(); for (int i = 0; i < size; ++i) { // 当前队列中有多少个元素则将依次放入array TreeNode * ptr = queue.front();//获得队列队首元素 array.push_back(ptr->val);//插入到该行数组中 if (ptr->left != nullptr) {// 左节点不为空,将其插入队列 queue.push(ptr->left); } if (ptr->right) {//右节点不为空,将其插入队列 queue.push(ptr->right); } queue.pop();//移除已经插入的节点 } a_array.push_back(array); } return a_array; } };
核心点在于利用队列先进先出的特点,先放入父节点,在出队的时候放入其节点的指向的子节点。实现层次遍历