第五屆天梯賽模擬題 - L2-3二叉搜尋樹的2層結點統計 (25分) - 建樹+DFS
阿新 • • 發佈:2020-11-23
思路
我感覺關於樹的題目我不能很好的寫出來!趕緊複習一下大部分樹的板子及變形題!!!
題意
給出 \(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; }