PAT (Advanced Level) 1004 Counting Leaves(树的遍历)
生活随笔
收集整理的這篇文章主要介紹了
PAT (Advanced Level) 1004 Counting Leaves(树的遍历)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目鏈接:點擊查看
題目大意:給出一個樹狀家譜,問每一代沒有后代的節點個數
題目分析:其實就是個簡單的樹的遍歷,奈何讀不懂題。。遍歷到每一層然后記錄沒有后代的個數即可,我是習慣性的建了無向圖,遍歷的時候沒有后代的條件是度為1并且前置父節點不是-1,然后需要對根節點特判一下,特判一下是否只有一個根節點,兩種代碼,都寫來練練手:
DFS:
#include<iostream> #include<cstdlib> #include<string> #include<cstring> #include<cstdio> #include<algorithm> #include<climits> #include<cmath> #include<cctype> #include<stack> #include<queue> #include<list> #include<vector> #include<set> #include<map> #include<sstream> #include<unordered_map> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=110;int ans[N];int mmax=-1;vector<int>node[N];void dfs(int u,int fa,int step) {if(node[u].size()==1&&fa!=-1||node[u].size()==0&&fa==-1){ans[step]++;mmax=max(mmax,step);return;}for(int i=0;i<node[u].size();i++){int v=node[u][i];if(v==fa)continue;dfs(v,u,step+1);} }int main() { // freopen("input.txt","r",stdin);int n,m;scanf("%d%d",&n,&m);while(m--){int u,v,num;scanf("%d%d",&u,&num);while(num--){scanf("%d",&v);node[u].push_back(v);node[v].push_back(u);}}dfs(1,-1,0);cout<<ans[0];for(int i=1;i<=mmax;i++)cout<<' '<<ans[i];return 0; }BFS:
#include<iostream> #include<cstdlib> #include<string> #include<cstring> #include<cstdio> #include<algorithm> #include<climits> #include<cmath> #include<cctype> #include<stack> #include<queue> #include<list> #include<vector> #include<set> #include<map> #include<sstream> #include<unordered_map> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=110;int ans[N];int mmax=-1;vector<int>node[N];struct Node {int to,step,fa;Node(int TO,int FA,int STEP){to=TO;fa=FA;step=STEP;} };void bfs() {queue<Node>q;q.push(Node(1,-1,0));while(!q.empty()){Node cur=q.front();q.pop();int u=cur.to;int fa=cur.fa;int step=cur.step;if(node[u].size()==1&&fa!=-1||node[u].size()==0&&fa==-1){ans[step]++;mmax=max(mmax,step);continue;}for(int i=0;i<node[u].size();i++){int v=node[u][i];if(v==fa)continue;q.push(Node(v,u,step+1));}} }int main() { // freopen("input.txt","r",stdin);int n,m;scanf("%d%d",&n,&m);while(m--){int u,v,num;scanf("%d%d",&u,&num);while(num--){scanf("%d",&v);node[u].push_back(v);node[v].push_back(u);}}bfs();cout<<ans[0];for(int i=1;i<=mmax;i++)cout<<' '<<ans[i];return 0; }?
總結
以上是生活随笔為你收集整理的PAT (Advanced Level) 1004 Counting Leaves(树的遍历)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: PAT (Advanced Level)
- 下一篇: PAT (Advanced Level)