二叉树的层次遍历用数组返回

最新推荐文章于 2023-06-16 14:50:37 发布

半梦半醒间幸运_

最新推荐文章于 2023-06-16 14:50:37 发布

阅读量761

点赞数

分类专栏： c++算法代码

本文链接：https://blog.csdn.net/weixin_43086349/article/details/104615691

版权

c++算法代码专栏收录该内容

114 篇文章 1 订阅

订阅专栏

二叉树的层次遍历
遍历结果每一层用数组保存

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)
  {
    vector<vector<int>> ans;
    if(!root) return NULL;
    queue<TreeNode*> q;
    q.push(root);
    while(!q.empty())
    {
       vector<int> temp;
       int len=q.size();
       for(int i=0;i!=len;i++)
       {
         TreeNode* t=q.front();
         temp.push_back(t->val);
         q.pop();
         if(t->left) q.push(t->left);
         if(t->right) q.push(t->right);
       }
       ans.push_back(temp);
    }
    return ans;
  }
}