2018天梯賽 L2-2 小字輩
阿新 • • 發佈:2019-01-06
L2-2 小字輩
本題給定一個龐大家族的家譜,要請你給出最小一輩的名單。
輸入格式:
輸入在第一行給出家族人口總數 N(不超過 100 000 的正整數) —— 簡單起見,我們把家族成員從 1 到 N 編號。隨後第二行給出 N 個編號,其中第 i 個編號對應第 i 位成員的父/母。家譜中輩分最高的老祖宗對應的父/母編號為 -1。一行中的數字間以空格分隔。
輸出格式:
首先輸出最小的輩分(老祖宗的輩分為 1,以下逐級遞增)。然後在第二行按遞增順序輸出輩分最小的成員的編號。編號間以一個空格分隔,行首尾不得有多餘空格。
輸入樣例:
9
2 6 5 5 -1 5 6 4 7
輸出樣例:
4
1 9
這題我在比賽沒有解出來真的是傻了(雖然也有一部分其他原因)
題解:
暴力深搜每一個人的祖先,然後每個進行記錄路徑(即當前的人的輩分是多少),然後因為是順序訪問,放進vector裡面肯定也是順序的,所以不用排序。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int pre[maxn];
vector<int>ans[maxn];
int aans[maxn];
int cnt=0;
int find(int x)
{
if(aans[x])
{
// cout<<"++++++++"<<endl;
return aans[x];
}
if(pre[x]==-1)
{
return aans[x]=1;
}
else
{
// cout<<"-------"<<endl;
aans[x]=find(pre[x])+1;
return aans[x];
}
}
int main()
{
int n;
ios::sync_with_stdio(false);
cin>>n;
int mmax=-1;
for (int i=0;i<=n;i++)
{
pre[i]==i;
}
for(int i=1;i<=n;i++)
{
cin>>pre[i];
}
for(int i=1;i<=n;i++)
{
int temp;
temp=find(i);
mmax=max(mmax,temp);
ans[temp].push_back(i);
}
cout<<mmax<<endl;
for(int i=0;i<ans[mmax].size();i++)
{
printf("%d",ans[mmax][i]);
if(i<ans[mmax].size()-1)
{
printf(" ");
}
else
{
printf("\n");
}
}
}