题解 | #求二叉树的层序遍历#

求二叉树的层序遍历

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;
    }
};

核心点在于利用队列先进先出的特点,先放入父节点,在出队的时候放入其节点的指向的子节点。实现层次遍历

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务