二叉树的层次遍历

最新推荐文章于 2023-10-12 22:04:51 发布

原创最新推荐文章于 2023-10-12 22:04:51 发布 · 225 阅读

0 ·

CC 4.0 BY-SA版权

数据结构—二叉树专栏收录该内容

16 篇文章

订阅专栏

给出一棵二叉树，返回其节点值的层次遍历（逐层从左往右访问）

您在真实的面试中是否遇到过这个题？

Yes

样例

给一棵二叉树 {3,9,20,#,#,15,7} ：

返回他的分层遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

解题思路：建立两个队列滚动数组进行循环每次统计一层的结点，当两个队列都为空时跳出循环。

class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
public:
    vector<vector<int>>ans;
    vector<vector<int>> levelOrder(TreeNode *root) {
         if(root==NULL)return ans;
         queue<TreeNode *>tem[2];
         tem[0].push(root);
         int i=0;
         while(!tem[0].empty()||!tem[1].empty()){
             vector<int>t;
             while(!tem[i].empty()){
                 t.push_back(tem[i].front()->val);
                 if(tem[i].front()->left!=NULL)tem[(i+1)%2].push(tem[i].front()->left);
                 if(tem[i].front()->right!=NULL)tem[(i+1)%2].push(tem[i].front()->right);
                 tem[i].pop();
             }
            ans.push_back(t);
             i=(i+1)%2;
         }
      return ans;
        
    }
};