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

求二叉树的层序遍历

https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
        vector<vector<int>> result; // 创建结果数组
        queue<TreeNode*> que;   // 创建单向队列,存放每层的结点
        if(root) que.push(root);    // 如果根节点不为空,将根节点放入队列中
        while(!que.empty()){    // 当队列不为空时,进入循环
            vector<int> temp_vec;   // 创建临时数组,保存每层结点值
            int deq_size = que.size();  // 保存每层结点数,此为每层遍历次数
            for(int i = 0; i < deq_size; ++i){  // 进入循环
                TreeNode* cur = que.front();    // 临时结点等于队头结点
                temp_vec.push_back(cur->val);   // 将队头结点值放入临时数组中
                que.pop();    // 删除队头结点
                if(cur->left) que.push(cur->left);    // 当左孩子不为空,将左孩子放入队列中
                if(cur->right) que.push(cur->right);  // 当右孩子不为空,将右孩子放入队列中
            }
            result.push_back(temp_vec); // 将该层结点值放入结果数组中
        }
        return result;
    }
};

全部评论

相关推荐

被加薪的哈里很优秀:应该继续招人,不会给你留岗位的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务