NC15 求二叉树的层序遍历

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

题目示例:

 该二叉树层序遍历的结果是 [ [3], [9,20], [15,7] ]

解题思路:层序遍历可以利用队列实现。首先将根节点加入队列,依次弹出队首,并将队首节点的左右子节点依次加队尾。依次循环,直至队列为空,即完成对二叉树的层序遍历。注意要设置一变量来判断遍历的层数。C++代码实现如下:

    vector<vector<int> > levelOrder(TreeNode* root) {
        // write code here
        vector<vector<int>> ans;
        if(!root)
            return ans;
        vector<int> an;
        queue<TreeNode *>sTree;
        sTree.push(root);
        while(!sTree.empty())
        {
            int sTreeSize  = sTree.size();
                while(sTreeSize>0)
                {
                    TreeNode *node = sTree.front();
                    sTree.pop();
                    sTreeSize--;
                    an.push_back(node->val);
                    if(node->left!=nullptr)
                        sTree.push(node->left);
                    if(node->right!=nullptr)
                        sTree.push(node->right);
                }
            ans.push_back(an);
            an.clear();
        }
        return ans;
    }

猜你喜欢

转载自blog.csdn.net/pigbossa/article/details/120469388