【PTA】小字辈

阳光穿透心脏的1/2处 2023-07-04 12:50 8阅读 0赞

#### 题目重述 ####

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

#### 输入格式： ####

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

#### 输出格式： ####

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

#### 输入样例： ####

9
    2 6 5 5 -1 5 6 4 7

#### 输出样例： ####

4
    1 9

#### 题解 ####

*  题目的意思是找到最小辈分,这里要注意最大辈分是1.
 *  在输入时就记录祖先结点,从祖先结点开始遍历，遍历的过程中，如果遍历时的辈分与全局变量记录的辈分一样，就用vector记录下来，因为题目就是让我们输出最小辈分.
 *  如果当前辈分小于全局变量记录的辈分，那就将vector清空，因为最小辈分变了，vector里之前记录的就没用了。

#### C++ AC ####

#include <iostream>
    #include <bits/stdc++.h>
    using namespace std;
    vector<int> store[100005],v;
    int maxdai=-1;
    //从祖先结点开始深度搜索
    void dfs(int root,int dai)
    {
        
        //如果搜索到的代就是最小辈分，就放v里面保存起来
        if(dai==maxdai)
        {
        
            v.push_back(root);
    
        }
        //如果搜索到的代大于最大备份，那之前保存的没什么用了，v清空
        if(dai>maxdai)
        {
        
            v.clear();
            maxdai=dai;
            v.push_back(root);
        }
        //深度搜索
        for(int i=0;i<store[root].size();i++)
        {
        
            dfs(store[root][i],dai+1);
        }
    }
    int main()
    {
        
        int n,temp,ans;
        cin>>n;
        for(int i=1;i<=n;i++)
        {
        
            cin>>temp;
            //找到辈分最高的祖先
            if(temp==-1)
            {
        
                ans=i;
            }
            //因为输入的是i编号的父辈，所以记录这个父辈的孩子都有谁
            else
            {
        
                store[temp].push_back(i);
            }
        }
        dfs(ans,1);
    
        sort(v.begin(),v.end());
        cout<<maxdai<<endl;
        cout<<v[0];
       for(int i=1;i<v.size();i++)
       {
        
           cout<<" "<<v[i];
       }
        return 0;
    }