7-11 二分探索木のレイヤー2ノード統計(C++)

二分探索木は、空の木、または次のプロパティを持つ二分木です。左の部分木が空でない場合、左の部分木のすべてのノードの値はルート ノードの値以下です。 ; 右のサブツリーが空でない場合、右のサブツリーのすべてのノードの値はルート ノードの値よりも大きく、左と右のサブツリーも二分探索木です。

最初は空の二分探索木に一連の数値を所定の順序で挿入します。タスクは、結果木の下部 2 層にあるノードの数を数えることです。

入力形式: 1 行目に正の整数 N (≤1000) を入力します。これは、挿入される数字の数です。2 行目は [−1000,1000] の区間で N 個の整数を与えます。数字はスペースで区切ります。

出力形式：下2層のノード総数を1行で出力します。

入力サンプル: 9
25 30 42 16 20 20 35 -5 28
出力サンプル: 6

コード長制限

16キロバイト

制限時間

400ミリ秒

メモリ制限

64MB

#include <bits/stdc++.h>
using namespace std;

typedef struct Node {
    int val;
    Node* left;
    Node* right;
    int size; //子树数量
    int tall;//高度
}node, *pNode;

int max1 = 0;
int daan = 0;

//创建二叉搜索树
pNode buildTree(pNode &tree, int num,int tall){

    if(tree == NULL){
        tree = new node;  //开辟空间
        tree->val = num;  //赋值
        tree->left = tree->right = NULL;
        tree->tall = tall;//深度
        if(tall > max1){ //记录最大深度
            max1 = tall;
        }
 //cout << "22" << ' ' << tree->val <<endl;
        return tree;
    }

    //用二层搜索树的性质递归插入结点
    if(num <= tree->val){
        tree->left = buildTree(tree->left, num, tall+1);
    }
    else {
        tree->right = buildTree(tree->right, num, tall+1);
    }

    return tree;
}

//中序遍历
void inOrder(pNode &root){
    if(root == NULL)
        return;

    inOrder(root->left);
    if(root->tall >= max1-1){  //如果是最下两层，答案daan+1
        daan++;
    }
    inOrder(root->right);

}



int main(){
    int n;
    cin >> n;
    int num;
    node *root = NULL;
    for(int i = 0; i < n; i++){
        cin >> num;
        buildTree(root, num, 1);
    }
    //cout << max1 << endl;
    inOrder(root);
    cout << daan;

    return 0;
}