L2-026 小字辈

太过爱你忘了你带给我的痛 2022-03-16 12:18 129阅读 0赞

L2-026 小字辈 （25 分）

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

### 输入格式： ###

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

### 输出格式： ###

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

### 输入样例： ###

9
    2 6 5 5 -1 5 6 4 7

### 输出样例： ###

4
    1 9

直接这样暴力深搜会超时，只有16分。

#include<stdio.h>
    #include<string.h>
    #include<math.h>
    #include<iostream>
    #include<string>
    #include<algorithm>
    #include<map>
    #include<set>
    #include<queue>
    #include<stack>
    #include<vector>
    using namespace std;
    #define inf 0x3f3f3f3f
    #define LL long long
    int a[100005];
    struct sb
    {
    	int i,bf;
    }b[100005];
    int dfs(int x,int bf)
    {
    	if(x==-1) return bf;
    	return dfs(a[x],bf+1);
    }
    bool cmp(sb u,sb w)
    {
    	if(u.bf!=w.bf)
    	return u.bf>w.bf;
    	else
    	return u.i<w.i;
    }
    int main()
    {
    	int n,i,j;
    	scanf("%d",&n);
    	for(i=1;i<=n;i++)	
    	scanf("%d",&a[i]);
    	for(i=1;i<=n;i++)
    	{
    		b[i].i=i;
    		b[i].bf=dfs(a[i],1);
    		//printf("%d %d\n",i,b[i]);
    	}
    	sort(b+1,b+1+n,cmp);
    	printf("%d\n",b[1].bf);
    	printf("%d",b[1].i);
    	for(i=2;i<=n;i++)
    	{
    		if(b[i].bf==b[1].bf)
    		printf(" %d",b[i].i);
    		else break;
    	}
    	printf("\n");
    }

优化一下， 用vector存一辈人。即在递归过程中，如果发现更小的一辈，那么就将v清空，并将这个人的编号添加到v中，如果发现同样是最小一辈的人，则将该人编号添加到v中。

#include<stdio.h>
    #include<string.h>
    #include<math.h>
    #include<iostream>
    #include<string>
    #include<algorithm>
    #include<map>
    #include<set>
    #include<queue>
    #include<stack>
    #include<vector>
    using namespace std;
    #define inf 0x3f3f3f3f
    #define LL long long
    vector<int> d[100005],v;
    int mbf=0;// max 辈分 
    void dfs(int x,int bf)
    {
    	if(bf==mbf) v.push_back(x);
    	if(bf>mbf)
    	{
    		mbf=bf;
    		v.clear();
    		v.push_back(x);
    	}
    	for(int i=0;i<d[x].size();i++)
    	dfs(d[x][i],bf+1);
    }
    int main()
    {
    	int n,i,j,a,root;
    	scanf("%d",&n);
    	for(i=1;i<=n;i++)	
    	{
    		scanf("%d",&a);
    		if(a==-1)
    		root=i;
    		else
    		d[a].push_back(i);
    	}
    	dfs(root,1);
    	sort(v.begin(),v.end());
    	printf("%d\n",mbf);
    	for(i=0;i<v.size();i++)
    	{
    		if(i)printf(" ");
    		printf("%d",v[i]);
    	}
    	printf("\n");
    }