小字辈 (bfs好题)

小字辈 (25 分)

本题给定一个庞大家族的家谱,要请你给出最小一辈的名单。

输入格式:

输入在第一行给出家族人口总数 N(不超过 100 000 的正整数) —— 简单起见,我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号,其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。

输出格式:

首先输出最小的辈分(老祖宗的辈分为 1,以下逐级递增)。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔,行首尾不得有多余空格。

输入样例:

9
2 6 5 5 -1 5 6 4 7

输出样例:

4
1 9
 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <vector>
 5 #include <queue>
 6 #include <algorithm>
 7 using namespace std;
 8 const int maxn=1e5+10;
 9 vector<int> v[maxn];
10 vector<int> f[maxn];
11 int n,x;
12 int high;
13 int parent;
14 void bfs()
15 {
16     queue<int>q;
17     q.push(parent);
18     high=0;
19     f[high].push_back(parent);
20 
21     while(!q.empty())
22     {
23 
24         int sum=q.size();
25         high++;
26         for(int i=0;i<sum;i++)
27         {
28             int head=q.front();
29             q.pop();
30             for(int j=0;j<v[head].size();j++)
31             {
32                 f[high].push_back(v[head][j]);
33                 q.push(v[head][j]);
34             }
35 
36         }
37     }
38 }
39 int main()
40 {
41     cin>>n;
42     for(int i=1;i<=n;i++)
43     {
44         scanf("%d",&x);
45         if(x>0)
46             v[x].push_back(i);
47         else
48             parent=i;
49     }
50     bfs();
51     cout<<high<<endl;
52     for(int i=0;i<f[high-1].size();i++)
53     {
54         if(i!=0)
55             printf(" ");
56 
57             printf("%d",f[high-1][i]);
58     }
59     cout<<endl;
60     return 0;
61 }

 第二种方法:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=1e5+10;
int n,x,parent;
struct node{
    int to;
    int nex;
}e[maxn];
int a[maxn];
int cnt;
int len;
int head[maxn],vis[maxn];
void init()
{
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
    cnt=0;
    return;
}
void add(int x,int y)
{
    e[cnt].to=y;
    e[cnt].nex=head[x];
    head[x]=cnt++;
    return;
}
vector<int>f[maxn];
int t=1;
int maxx=0;
void bfs()
{
    priority_queue<int,vector<int>,greater<int> > q;
    len=1;
    q.push(parent);
    vis[parent]=t;
    f[vis[parent]].push_back(parent);
    while(!q.empty())
    {
        int v=q.top();
        q.pop();
        len++;
        for(int i=head[v];i!=-1;i=e[i].nex)
        {
            int u=e[i].to;
            vis[u]=vis[v]+1;
         //   cout<<u<<" "<<vis[u]<<endl;
            f[vis[u]].push_back(u);
            maxx=max(maxx,vis[u]);
            q.push(u);
        }
    }
}
int main()
{
    cin>>n;
    init();
    int xx;
    if(n==1)
        cin>>xx,
        printf("1
1
");
    else
    {
        for(int i=1;i<=n;i++)
        {
            cin>>x;
            if(x>0)
            {
                add(x,i);
            }
            else
            {
                parent=i;
            }
        }
       bfs();
       // cout<<maxx<<endl;
        cout<<maxx<<endl;
        int m=0;
        for(int i=0;i<f[maxx].size();i++)
        {
            a[i]=f[maxx][i];
            m++;
        }
        sort(a,a+m);
        for(int i=0;i<m;i++)
        {
            if(i)
                printf(" ");
            printf("%d",a[i]);
        }
    }

    return 0;
}
原文地址:https://www.cnblogs.com/1013star/p/9750803.html