思路

我感覺關於樹的題目我不能很好的寫出來！趕緊複習一下大部分樹的板子及變形題！！！

題意

給出 \(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;
}