打印二叉树每一层节点数量

最新推荐文章于 2021-10-27 20:21:18 发布

brucehb

最新推荐文章于 2021-10-27 20:21:18 发布

阅读量2.5k

点赞数 2

分类专栏：算法

本文链接：https://blog.csdn.net/brucehb/article/details/8723939

版权

算法专栏收录该内容

636 篇文章 2 订阅

订阅专栏

#include <iostream>
#include <queue>
using namespace std;

struct Node
{
      int value;
      Node *left;
      Node *right;
};

void printNumOfEveryLevel(Node *root)
{
      if (NULL == root)
      {
          return;
      }
      queue<Node*> buf;
      buf.push(root);
      buf.push(NULL);
      int count = 0;
      int level = 0;
      while (buf.size())
      {
          Node *front = buf.front();
          buf.pop();
          if (front == NULL)
          {
              level++;
              cout << level << " : " << count << endl;
              count = 0;
              if (buf.empty())
              {
                  break;
              }
              else
              {
                  buf.push(NULL);
              }
          }
          else
          {
              count++;
              if (front->left != NULL)
              {
                  buf.push(front->left);
              }
              if (front->right != NULL)
              {
                  buf.push(front->right);
              }
          }
      }
}