102- Binary Tree Level Order Traversal

类别:BFS
难度:medium

题目描述

这里写图片描述

算法分析

(1)广度优先算法,使用队列辅助实现
(2)如果单纯是从左到右,逐层遍历二叉树的话,就很简单:
最初将根节点压入队列中,在队列不为空的情况下,不断的从队头取出元素,并且将队头元素的左右节点压入队列中,知道队列为空,则遍历结束。
(4)这道题目与普通的遍历存在一个不同的地方是:需要按层进行分析,所以使用了一个level作为标记,并且压入队列中的除了节点以外附加了level的信息,这样当从队列中取出元素的时候如果是旧的level,则直接压入数组中,如果是新的level,则将原来那一层先存在result中,然后删除数组,并将新的节点放入数组中,开始新的一层的数组。
(5)需要注意的一点是,在进行vector的erase操作的时候,因为使用的是iterator,从头往后进行删除,每次删除完了以后,后面的数字就往前移动了,所以it不需要++,如果在for中使用it++,则在内部实现中需要使用it–.

代码实现

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        if (root == NULL) return result;
        queue<pair<TreeNode*, int> > que;
        que.push(make_pair(root, 0));
        int curlevel = -1;
        int level = 0;
        pair<TreeNode*, int> curNode;
        vector<int> nodes;
        // BFS ---- using queue
        while (!que.empty()) {
            curNode = que.front();
            que.pop();
            if (curNode.second != curlevel) {
                curlevel = curNode.second;
                if (nodes.size() != NULL) {
                    result.push_back(nodes);
                }
                // notice no it++
                for (auto it = nodes.begin(); it != nodes.end();) {
                    nodes.erase(it);
                }
            }
            nodes.push_back(curNode.first->val);
            if (curNode.first->left != NULL) {
                que.push(make_pair(curNode.first->left, curlevel + 1));
            }
            if (curNode.first->right != NULL) {
                que.push(make_pair(curNode.first->right, curlevel + 1));
            }
        }
        if (!nodes.empty()) result.push_back(nodes);
        return result;     
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值