二叉搜索树的2层结点统计 (25 分)

二叉搜索树的2层结点统计 (25 分)

 

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

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

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

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

输入样例：9 25 30 42 16 20 20 35 -5 28 输出样例：6

---

#include<bits/stdc++.h>
using namespace std;
int l[1005],r[1005],a[1005];
int num[1005],max_d,n,x;
void dfs(int root,int pos,int depth){
    if(a[pos] <= a[root]){
        if(l[root]==0x3f3f3f3f){
            l[root]=pos;
            num[depth+1]++;
            max_d=max(max_d,depth+1);
        }else{
            dfs(l[root],pos,depth+1);

        }
    }else{
        if(r[root]==0x3f3f3f3f){
            r[root]=pos;
            num[depth+1]++;
            max_d=max(max_d,depth+1);
        }else{
            dfs(r[root],pos,depth+1);
        }
    }
}
int main(){
    memset(l,0x3f3f3f3f,sizeof l);
    memset(r,0x3f3f3f3f,sizeof r);

    cin>>n;
    if(n==1){
        cout<<"1"<<endl;
        return 0;
    }
    for(int i=0;i<n;i++){
        cin>>x;
        if(!i) a[i]=x;
        else{
            a[i]=x;
            dfs(0,i,0);
        }
    }
    cout<<num[max_d-1]+num[max_d];
    return 0;
}