PAT_甲级_1115 Counting Nodes in a BST

题目大意

给定一颗二叉搜索树的插入序列，计算最后两层的节点个数

算法思路

首先我们将这n个数字依次插入到二叉搜索树中，然后使用层序遍历获取每一层节点的数目和最大层数maxLevel，L[maxLevel],L[maxLevel-1]就是最后一层和倒数第二层的节点个数

提交结果

AC代码

#include
#include
#include

using namespace std;

struct Node{
    int val;
    Node* left;
    Node* right;
    int level;
};

/// 将x插入到root中
void insert(Node* &root,int x){
    if(root== nullptr){
        root = new Node;
        root->val = x;
        root->left = nullptr;
        root->right = nullptr;
        return;
    }
    if(root->valright,x);
    }else{
        insert(root->left,x);
    }
}

// 根据数组arr建树
Node* createTree(int arr[],int n){
    Node* root = nullptr;
    for(int i=0;i que;
    root->level = 1;
    que.push(root);
    while(!que.empty()){
        Node* t = que.front();
        que.pop();
        maxLevel = max(maxLevel,t->level);
        ++L[t->level];
        if(t->left){
            t->left->level = t->level + 1;
            que.push(t->left);
        }
        if(t->right){
            t->right->level = t->level + 1;
            que.push(t->right);
        }
    }
}

int main() {
    int n;
    scanf("%d",&n);
    int initial[n];
    for(int i=0;i