PAT 二叉树打卡

acwing:1605. 二叉搜索树最后两层结点数量

二叉搜索树 (BST) 递归定义为具有以下属性的二叉树:

  • 若它的左子树不空,则左子树上所有结点的值均小于或等于它的根结点的值
  • 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值
  • 它的左、右子树也分别为二叉搜索树

将一系列数字按顺序插入到一个空的二叉搜索树中,然后,请你计算结果树的最低两层的结点个数。

输入格式

第一行包含整数 NN,表示插入数字序列包含的数字个数。

第二行包含 NN 个整数,表示插入数字序列。

输出格式

以如下格式,在一行中,输出结果树的最后两层的结点数:

复制代码
n1 + n2 = n

n1 是最底层结点数量,n2 是倒数第二层结点数量,n 是它们的和。

数据范围

1≤N≤10001≤N≤1000,

−1000≤−1000≤ 插入数字 ≤1000≤1000。

输入样例:
复制代码
9
25 30 42 16 20 20 35 -5 28
输出样例:
复制代码
2 + 4 = 6
cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=1010;

int n;

struct tree{
    int val;
    int h;
    int l,r;
}p[N];
int max_h;
int cnt[N];
void build(int root,int u,int h){
    if(p[root].val>=p[u].val){
        if(p[root].l==0x3f3f3f3f){
            p[root].l=u;
            p[u].h=h+1;
            max_h=max(max_h,p[u].h);
            //cout<<max_h<<endl;
            cnt[p[u].h]++;
        }else{
            build(p[root].l,u,h+1);
        }
    }else{
        if(p[root].r==0x3f3f3f3f){
            p[root].r=u;
            p[u].h=h+1;
            max_h=max(max_h,p[u].h);
           // cout<<max_h<<endl;
            cnt[p[u].h]++;
        }else{
            build(p[root].r,u,h+1);
        }
    }
}
void init()
{
    for(int i=0;i<N;i++){
        p[i].l=p[i].r=0x3f3f3f3f;
    }
} 
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>n;
    init();
    if(n==1){
        cout<<"1 + 0 = 1";
    }else{
        for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        if(i==1){
            p[1].val=x;
            p[1].h=1;
        }else{
            p[i].val=x;
            build(1,i,1);
        }
       
    }
    //cout<<max_h<<endl;
    cout<<cnt[max_h]<<" + "<<cnt[max_h-1]<<" = "<<cnt[max_h]+cnt[max_h-1];
    }
    
}
相关推荐
咸鱼爱学习9 小时前
【题解】B2613【深基1.习5】打字速度
数据结构·c++·算法
阿林学习计算机9 小时前
AVL树的实现
数据结构
whm277710 小时前
Visual Basic 值传递与地址传递
java·开发语言·数据结构
立志成为大牛的小牛10 小时前
数据结构——二十二、并查集(王道408)
c语言·数据结构·笔记·学习·考研
云青黛10 小时前
轮廓系数(一个异型簇的分类标准)
人工智能·算法·机器学习
云青黛10 小时前
肘部法找k
人工智能·算法·机器学习·聚类
CHANG_THE_WORLD10 小时前
if条件语句 三目运算符 汇编分析
汇编·算法·条件语句·if 语句·汇编分析·条件语句汇编分析
tumu_C10 小时前
无用知识研究:在trailing return type利用decltype,comma operator在对函数进行sfinae原创 [二]
开发语言·c++·算法
web安全工具库11 小时前
告别刀耕火种:用 Makefile 自动化 C 语言项目编译
linux·运维·c语言·开发语言·数据库·算法·自动化
海蓝可知天湛11 小时前
利用Genspark自定义智能体:算法竞赛测试数据反推与生成工具
算法·aigc