【GPLT 二阶题目集】L4-202 二叉搜索树的2层结点统计

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

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

输入格式：

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

输出格式：

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

输入样例：

9
25 30 42 16 20 20 35 -5 28

输出样例：

6

#include <iostream>
#include <algorithm>
#define MAXSIZE 10000010 //仍然不够
using namespace std;

//int tree[MAXSIZE]; //用数组需要的内存特别大，改用链表
typedef struct node {
    int key;
    node* lchild;
    node* rchild;
}node;
int deep[MAXSIZE] = { 0 }; //某深度的结点数
int num, de_max = 1, res = 0;

//当前节点的下标，当前深度，上一结点的下标
void dfs(node*& x, int de)
{
    if (x == NULL)
    {
        x = new node(); //建立新结点
        x->key = num;
        deep[de]++;
        de_max = max(de_max, de); //更新最大深度
    }
    else
    {
        if (num <= x->key) //按照二叉搜索树定义确定方向
            dfs(x->lchild, de + 1);
        else
            dfs(x->rchild, de + 1);
    }
    return;
}

int main()
{
    int n; cin >> n;
    if (n == 1) //结点数为1直接输出结果
    {
        cout << 1 << endl;
        return 0;
    }
    cin >> num; //创建根结点
    node* root = new node();
    root->key = num;
    for (int i = 0; i < n - 1; i++)
    {
        cin >> num;
        dfs(root, 1);
    }
    cout << deep[de_max] + deep[de_max - 1] << endl;
    return 0;
}

 注意事项：

一开始以为1000个结点很少，所以打算用数组，后来发现两个测试点段错误，最坏情况是深度1000，结点数2^1000-1个，太可怕了。

如有问题，欢迎提出。