计算二叉树的深度,结点的个数。

计算二叉树的深度,结点的个数。

//算法5.5 计算二叉树的深度,结点的个数。
#include<iostream>
using namespace std;

//二叉树的二叉链表存储表示
typedef struct BiNode
{        
  char data;            //结点数据域
  struct BiNode *lchild,*rchild;  //左右孩子指针
}BiTNode,*BiTree;

//用算法5.3建立二叉链表
void CreateBiTree(BiTree &T)
{  
  //按先序次序输入二叉树中结点的值（一个字符），创建二叉链表表示的二叉树T
  char ch;
  cin >> ch;
  if(ch=='#')  T=NULL;      //递归结束，建空树
  else{              
    T=new BiTNode;
    T->data=ch;          //生成根结点
    CreateBiTree(T->lchild);  //递归创建左子树
    CreateBiTree(T->rchild);  //递归创建右子树
  }                //else
}                  //CreateBiTree

int Depth(BiTree T)
{ 
  int m,n;
  if(T == NULL ) return 0;        //如果是空树，深度为0，递归结束
  else 
  {              
    m=Depth(T->lchild);      //递归计算左子树的深度记为m
    n=Depth(T->rchild);      //递归计算右子树的深度记为n
    if(m>n) return(m+1);    //二叉树的深度为m 与n的较大者加1
    else return (n+1);
  }
}
int NodeCount(BiTree T)
{
  if(T==NULL)return 0;
  else return NodeCount(T->lchild)+NodeCount(T->rchild)+1;
}
void main()
{
  BiTree tree;
  cout<<"请输入建立二叉链表的序列：\n";
  CreateBiTree(tree);
  cout<<"数的深度为："<<Depth(tree)<<endl;
  cout<<"结点的个数为： "<<NodeCount(tree)<<endl;
}