求二叉树的层序遍历

求二叉树的层序遍历(牛客网)
题目:给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)

例如:
给定的二叉树是{3,9,20,#,#,15,7},该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
        // write code here
        vector<vector<int>> res;
        if (root == NULL)
            return res;
        queue<TreeNode *> tree;
        tree.push(root);
        int curNum = 1, nextNum = 0;
        vector<int> cow;
        while (!tree.empty()) {
            TreeNode *temp = tree.front();
            tree.pop();
            curNum--;
            cow.push_back(temp->val);
            if (temp->left) {
                tree.push(temp->left);
                nextNum++;
            }
            if (temp->right) {
                tree.push(temp->right);
                nextNum++;
            }
            if (curNum == 0) {
                res.push_back(cow);
                curNum = nextNum;
                nextNum = 0;
                cow.erase(cow.begin(), cow.end());
            }
        }
        return res;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值