输入一棵二叉树，求树的深度

最新推荐文章于 2021-04-29 15:53:05 发布

weixin_34297300

最新推荐文章于 2021-04-29 15:53:05 发布

阅读量171

点赞数

文章标签：数据结构与算法 python

原文链接：https://my.oschina.net/u/1423127/blog/277235

版权

2019独角兽企业重金招聘Python工程师标准>>>

题目描述：
输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。
- 样例输出：
- ```
2
```
- 样例输入：
- ```
32 3-1 -1-1 -1
```
- 输出：
- 输出一个整型，表示树的深度。
- 输入：
- 第一行输入有n，n表示结点数，结点号从1到n。根结点为1。 n <= 10。
  接下来有n行，每行有两个个整型a和b，表示第i个节点的左右孩子孩子。a为左孩子，b为右孩子。当a为-1时，没有左孩子。当b为-1时，没有右孩子。

#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <string.h>
#include <queue>
using namespace std;
#define MAX(a,b) ((a)>(b)?(a):(b))
typedef struct _BiNode
{
   int data;
   struct _BiNode *lchild;
   struct _BiNode *rchild;
} BiNode;
queue <BiNode *> Que;
BiNode* BuildBiTree(int len)
{
    int a, b, i;
    BiNode *root = new BiNode;
    root->data = 1;
    root->lchild = NULL;
    root->rchild = NULL;
    
    if (root == NULL)
    {
        return NULL;
    }
    Que.push(root);
    for (i = 1; i <= len; ++i)
    {
      cin>>a>>b;
      BiNode *pNode = Que.front();
      Que.pop();
      pNode->data = i;
      if (a != -1)
      {
          pNode->lchild = new BiNode;
          pNode->lchild->data = a;
          Que.push(pNode->lchild);
      }
      else
      {
          pNode->lchild = NULL;
      }
      if (b != -1)
      {
          pNode->rchild = new BiNode;
          pNode->rchild->data = b;
          Que.push(pNode->rchild);
      }
      else
      {
          pNode->rchild = NULL;
      }
    }
    return root;
   
}

void DestroyBiTree(BiNode **root)
{
   if (*root)
   {
      if ((*root)->lchild)
      {
         DestroyBiTree(&(*root)->lchild);
      }
      if ((*root)->rchild)
      {
         DestroyBiTree(&(*root)->rchild);
      }
      delete *root;
      *root = NULL;
   }
}

int CalculateBiTreeDepth(BiNode *root)
{
    if (root == NULL)
    {
       return 0;
    }
    else
      return MAX(CalculateBiTreeDepth(root->lchild),CalculateBiTreeDepth(root->rchild))+1;
       
}

int main()
{
   int  n, i, j;
   BiNode *pRoot;
   char input[100];
   while(cin>>n)
   {
      pRoot = BuildBiTree(n);
      cout<<CalculateBiTreeDepth(pRoot)<<endl;
      DestroyBiTree(&pRoot);
   }  
   return 0;
}

转载于:https://my.oschina.net/u/1423127/blog/277235