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

求二叉树的层序遍历

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

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务