1. 程式人生 > >2018天梯賽 L2-2 小字輩

2018天梯賽 L2-2 小字輩

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"); } } }