A1115 Counting Nodes in a Binary Search Tree

最新推荐文章于 2023-11-05 11:28:38 发布

NEFU AB-IN

最新推荐文章于 2023-11-05 11:28:38 发布

阅读量284

点赞数

分类专栏： PTA 文章标签：深度优先算法

本文链接：https://blog.csdn.net/qq_45859188/article/details/126842420

版权

PTA 专栏收录该内容

71 篇文章 1 订阅

订阅专栏

Link

文章目录

A1115 Counting Nodes in a Binary Search Tree

A1115 Counting Nodes in a Binary Search Tree

题意

Insert a sequence of numbers into an initially empty binary search tree. Then you are supposed to count the total number of nodes in the lowest 2 levels of the resulting tree.
思路

二叉树的建树与插入，牢记函数即可

代码

/*
* @Author: NEFU AB-IN
* @Date: 2022-09-13 20:50:59
* @FilePath: \GPLT\A1115\A1115.cpp
* @LastEditTime: 2022-09-13 21:18:02
*/
#include <bits/stdc++.h>
using namespace std;
#define N n + 100
#define SZ(X) ((int)(X).size())
#define IOS                                                                                                            \
    ios::sync_with_stdio(false);                                                                                       \
    cin.tie(nullptr);                                                                                                  \
    cout.tie(nullptr)
#define DEBUG(X) cout << #X << ": " << X << '\n'
typedef pair<int, int> PII;

#undef N
const int N = 1e5 + 10;

// #undef int

signed main()
{
    IOS;
    int n;
    cin >> n;
    vector<int> l(N), r(N), v(N);
    // BST 结点编号从1开始
    int root = 0, idx = 0;

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

    for (int i = 0; i < n; ++i)
    {
        int x;
        cin >> x;
        insert(root, x);
    }

    int mx = 0;
    vector<int> ans(N);
    function<void(int, int)> dfs = [&](int u, int dep) {
        if (!u)
            return;
        mx = max(mx, dep);
        ans[dep]++;
        dfs(l[u], dep + 1);
        dfs(r[u], dep + 1);
    };

    dfs(root, 0);

    printf("%d + %d = %d", ans[mx], ans[mx - 1], ans[mx - 1] + ans[mx]);
    return 0;
}