PTA天梯赛-二叉搜索树的2层结点统计

二叉搜索树或者是一棵空树，或者是具有下列性质的二叉树：若它的左子树不空，则左子树上所有结点的值均小于或等于它的根结点的值；若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；它的左、右子树也分别为二叉搜索树。

将一系列数字按给定顺序插入一棵初始为空的二叉搜索树，你的任务是统计结果树中最下面 2 层的结点数。

输入格式：输入在第一行给出一个正整数 N (≤1000)，为插入数字的个数。第二行给出 N 个 [−1000,1000] 区间内的整数。数字间以空格分隔。

输出格式：在一行中输出最下面 2 层的结点总数。

输入样例：9 25 30 42 16 20 20 35 -5 28 输出样例：6
思路:根据题目看得出来用到树的结构，并且搜索二叉树的左右子树也是二叉搜索树，这里我们就能想到建立树的过程中递归的思想。我们来看代码：

#include <iostream>
#include <stdlib.h> 
using namespace std;
typedef struct TNode  //创建树的结构 
{ int data;
  struct TNode *lchild;
  struct TNode *rchild;
}*BiTree;
int count=0,maxc=0;//下面两层节点数和最大次数 

//建立二叉搜索树 
int Create_Tree(BiTree &T,int t,int c) //树节点  传入插入的数据  初始层数 
{ 
  if(T==NULL) //如果树为空 插入 
   {T=(BiTree)malloc(sizeof(TNode));
    T->data=t;
    T->lchild=NULL;
    T->rchild=NULL;
     if(maxc<c)//更新最大层数
	  maxc=c;
	return 0;    
   }
   else if(t<=T->data) //小于等于找左边 
   {   Create_Tree(T->lchild,t,c+1);
   }
   else //大于找右边 
   { Create_Tree(T->rchild,t,c+1);
   }
  return 1;
}
int PreOrder(BiTree T,int c)  //先序遍历树 
{ if(T==NULL)
   return 0;
   if(c>=maxc-1) //如果到下面两层 
    count++; //节点数+1 
  PreOrder(T->lchild,c+1);
  PreOrder(T->rchild,c+1);
}
int main()
{ int i,N,t;
  BiTree T=NULL; //初始为空树 
  cin>>N;
  for(i=1;i<=N;i++)
	{ cin>>t;
	  Create_Tree(T,t,1);
	}
	PreOrder(T,1); 
	cout<<count<<endl;
	return 0;
}