第五届天梯赛模拟题 - L2-3二叉搜索树的2层结点统计 (25分) - 建树+DFS

思路

我感觉关于树的题目我不能很好的写出来！赶紧复习一下大部分树的板子及变形题！！！

题意

给出 $n$ 个节点数，在一行中输出最下面 $2$ 层的结点总数。

AC代码

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
typedef long long ll;
#define f(i,x,y) for(int i=x;i<y;i++)
#define sc(x) scanf("%d",&x);
#define scc(x,y) scanf("%d %d",&x,&y)
#define pr(x) printf("%d\n",x)

const int N=1010;

int l[N],r[N],w[N],id,cnt[N],ma;
//lr子节点的下标 w对应节点的权值 cnt每个深度的节点数，ma最大深度

void build(int &x, int ww)  //  给插入的权值x进行编号 + 保存x点权值
{
    if (!x)
        x=++id,w[x]=ww;
    else
    {
        if (ww<=w[x]) build(l[x],ww); //左子树上所有结点的值均<=它的根结点的值
        else build(r[x],ww); //右子树上所有结点的值均>它的根结点的值
    }
}

void dfs(int x, int depth)
{
    if (!x) return;
    cnt[depth]++; // 记录最后一行是几层（最大深度）
    ma=max(ma,depth);
    dfs(l[x],depth+1);
    dfs(r[x],depth+1);
}

int main()
{
    int n,rt=0; // 不能是1
    cin>>n;
    f(i,0,n)
    {
        int xx;
        cin>>xx;
        build(rt,xx);
    }
    dfs(rt,0);
    cout<<cnt[ma]+cnt[ma-1]<<endl;//最后一层+倒数第二层
    return 0;
}