【leetcode】Binary Tree Level Order Traversal

最新推荐文章于 2021-11-17 20:57:12 发布

原创最新推荐文章于 2021-11-17 20:57:12 发布 · 474 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 2 个专栏收录

105 篇文章

订阅专栏

leetcode

79 篇文章

订阅专栏

本文介绍了一种使用队列实现的二叉树层次遍历算法，并提供了详细的C++代码实现。通过该算法，可以有效地获取二叉树的每一层节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:

    vector<vector<int> > levelOrder(TreeNode *root) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        vector<vector<int> >ans;
        if(root==NULL)
            return ans;
            
        vector<int> vec;//每一层的节点放入里面
        queue<TreeNode *> que;     
        que.push(root);
        int count=1;//上一层的在队列中的节点数
        while(que.empty()==false)
        {
            vec.clear();
            int nextCount=0;//接下来的下一层的节点数的统计
            for(int i=0;i<count;i++)//弹出上一层的节点进行处理
            {
                TreeNode *curNode=que.front();
                que.pop();
            
                vec.push_back(curNode->val);//
            
                if(curNode->left!=NULL)
                {
                    que.push(curNode->left);
                    nextCount++;
                }
                if(curNode->right!=NULL)
                {
                    que.push(curNode->right);    
                    nextCount++;
                }
            }
            count=nextCount;
            ans.push_back(vec);
        }
        return ans;
    }
};