0
点赞
收藏
分享

微信扫一扫

【PAT甲级】1115 Counting Nodes in a BST(按序插入节点构造二叉树,输出最后两层的节点个数)

高子歌 2022-04-03 阅读 61

本题要点在于按照题目所给序列依次插入节点构造二叉树。构造二叉树采用l[x]和r[x]分别代表节点x的左右儿子的下标,若为0则说明该节点没有左/右儿子。
以1号节点为根节点,每次插入都从根节点开始,权值小于等于当前节点则前往左子树,否则前往右子树,直至找到一个空节点插入权值。最后dfs遍历一下所有节点,cnt记录下每一层的节点个数,输出最后两层即可。
需要注意的是insert函数中 int &u 实现了l[u]和r[u]的赋值。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;
int l[N], r[N], w[N];
int n, n1, n2;
int idx;
int cnt[N], maxDepth;

void insert(int &u, int weight)
{
    if(!u)
    {
        u = ++idx;
        w[u] = weight;
    }
    else if(weight <= w[u]) insert(l[u], weight);
    else insert(r[u], weight);
}

void dfs(int u, int depth)
{
    if(!u) return;
    cnt[depth] ++;
    maxDepth = max(depth, maxDepth);
    dfs(l[u], depth + 1);
    dfs(r[u], depth + 1);
}

int main()
{
    cin >> n;
    int root = 0;
    while(n --)
    {
        int t;
        cin >> t;
        insert(root, t);
    }
    dfs(root, 0);
    n1 = cnt[maxDepth], n2 = cnt[maxDepth - 1];
    printf("%d + %d = %d", n1, n2, n1 + n2);
}

举报

相关推荐

0 条评论