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

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

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

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

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

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

题目分析： 题目要求解的是二叉搜索树的最后两层的结点数，用dfs深度优先搜索，并且更新深度，用一个数组存放数据，再用两个数组存放根结点的左孩子和右孩子结点。

题目解答：

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
int pos[1001],cnt[1001],l[1001],r[1001];
int max_depth=0;
void dfs(int root,int u,int depth){
    if(pos[u]<=pos[root]){
        if(l[root]==INF){
            l[root]=u;
            cnt[depth+1]++;
            max_depth=max(max_depth,depth+1);
            return ;
        }else{
            dfs(l[root],u,depth+1);
            return ;
        }
    }else{
        if(r[root]==INF){
            r[root]=u;
            cnt[depth+1]++;
            max_depth=max(max_depth,depth+1);
            return ;
        }else{
            dfs(r[root],u,depth+1);
            return ;
        }
    }
}
int main(){
    int n;
    cin>>n;
    memset(l,0x3f,sizeof(l));
    memset(r,0x3f,sizeof(r));
    if(n==1){
        cout<<"1"<<endl;
        return 0;
    }
    for(int i=0;i<n;i++){
        int x;
        cin>>x;
        if(!i){
            pos[i]=x;
        }else{
            pos[i]=x;
            dfs(0,i,0);
        }
    }
    cout<<cnt[max_depth-1]+cnt[max_depth]<<endl;
    return 0;
}

 参考博客：二叉搜索树的2层结点统计_zhbbbb！的博客-CSDN博客_二叉搜索树的2层结点统计