PAT甲级1004 Counting Leaves (30分):[C++题解]树、邻接表存储树、dfs遍历树
生活随笔
收集整理的這篇文章主要介紹了
PAT甲级1004 Counting Leaves (30分):[C++题解]树、邻接表存储树、dfs遍历树
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
- 題目分析
- 題目鏈接
題目分析
題意重述:一棵樹,求每一層的葉子節點數目。
分析 構造樹,使用鄰接表來存(相當于存儲有向圖)。
需要一個頭結點數組h[N],然后每個頭節點往外形成一個單鏈表e[],ne[];(用數組模擬)單鏈表中用來存該結點的孩子結點,用cnt[N]數組來記錄每一層的葉子結點數。
使用深度優先遍歷dfs來求每一層的葉子節點數,由于是鄰接表式的存儲,頭結點數組中某結點值是-1,表示是葉子結點,這也就是dfs的退出的條件,此時如何退出呢? 需要將該葉子結點所在層數的葉子數++,然后記得更新樹的最大深度。
然后深度遍歷該結點的孩子結點。
ac代碼
#include<bits/stdc++.h> using namespace std;const int N=110; int n,m; //用鄰接表來存 int h[N], e[N],ne[N], idx; //idx默認初始化為0int cnt[N] ;// 每一層的葉子結點數int max_depth; //樹的最大層數void add(int a, int b){ //使用頭結點a的鏈表 h[a]//在a這個單鏈表內,插入結點b(頭插法)//表示a有一個兒子b//e[ ] 存的是結點號b//ne [ ] 中存的是下一個結點序號idx,不是結點號e[idx] =b,ne[idx] =h[a],h[a]=idx++;//cout<< a<<' '<<b<<endl; }void dfs(int u, int depth){if(h[u]==-1){ //說明u是葉子結點cnt[depth]++;//depth這一層的計數++max_depth=max(max_depth,depth);return;}//遍歷頭結點u對應的單鏈表h[u]for(int i =h[u]; i!=-1; i=ne[i]){dfs(e[i],depth+1);//}}int main(){cin>> n>> m;memset(h ,-1,sizeof h);for(int i=0;i<m;i++){int id ,k;cin>> id >> k;while(k--){int son;cin>>son;add(id , son);//構造鄰接表}}dfs(1,0); //結點號,層數cout<<cnt[0];for(int i=1;i<=max_depth;i++) cout<<" "<<cnt[i];}題目鏈接
PAT甲級1004 Counting Leaves (30分)
《新程序員》:云原生和全面數字化實踐50位技術專家共同創作,文字、視頻、音頻交互閱讀總結
以上是生活随笔為你收集整理的PAT甲级1004 Counting Leaves (30分):[C++题解]树、邻接表存储树、dfs遍历树的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 堆排序(如何手写堆)
- 下一篇: PAT甲级1020 Tree Trave