【PAT甲级】1115 Counting Nodes in a BST (30分)（二叉查找树）

题意：

输入一个正整数N（<=1000），接着输入N个整数（[-1000,1000]），依次插入一棵初始为空的二叉排序树。输出最底层和最底层上一层的结点个数之和，例如x+y=x+y。

AAAAAccepted code：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
typedef struct Node{
    int value;
    int vis;
    int level;
    int visl,visr;
    Node *lchild,*rchild;
}node;
int a[1007];
int num[1007];
void insert_(node *now,int x,int y){
    if(now->vis==0){
        now->value=x;
        now->vis=1;
        now->level=y;
        ++num[y];
        return ;
    }
    if(x<=now->value){
        if(now->visl==0){
            now->lchild=new node();
            now->visl=1;
        }
        insert_(now->lchild,x,1+y);
    }
    else{
        if(now->visr==0){
            now->rchild=new node();
            now->visr=1;
        }
        insert_(now->rchild,x,1+y);
    }
    return ;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    node *root=new node();
    root->value=a[1];
    root->vis=1;
    root->level=0;
    ++num[0];
    for(int i=2;i<=n;++i)
        insert_(root,a[i],0);
    int mx=0;
    for(int i=0;i<n;++i)
        if(num[i])
            mx=i;
    cout<<num[mx]<<" + "<<num[mx-1]<<" = "<<num[mx]+num[mx-1];
    return 0;
}