LeetCode: Binary Tree Level Order Traversal

原创于 2014-07-01 10:28:36 发布 · 582 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种使用双队列实现的二叉树层次遍历算法，通过交替使用两个队列来保存不同层级的节点，实现了对二叉树的层次遍历并返回每一层的节点值。

思路：使用两个队列，一个队列存储当前一层的所有节点，另一个队列存储下一层的节点，结合层次遍历树结构。

code:

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
        vector<vector<int> > ret;
        if(root!=NULL){
            TreeNode * temp = root;
            queue<TreeNode *> firstQ,nextQ;
            firstQ.push(temp);
            while(!firstQ.empty()){
                vector<int> layerRet;
                while(!firstQ.empty()){
                    TreeNode *t = firstQ.front();
                    if(t->left != NULL)
                        nextQ.push(t->left);
                    if(t->right != NULL)
                        nextQ.push(t->right);
                        
                    layerRet.push_back(t->val);
                    firstQ.pop();
                }
                firstQ = nextQ;
                while(!nextQ.empty())  
                    nextQ.pop();
                ret.push_back(layerRet);
            }
        }
        return ret;
    }
};