PTA L2-038 病毒溯源 (25 分)

PTA L2-038 病毒溯源 (25 分)

将所有节点的子节点从小到大排序，在选择的时候只会更长，且字典序最小的更新。

#include<bits/stdc++.h>
using namespace std;

const int N = 10010;

vector<int> g[N]; 
bool st[N];

vector<int> res;
vector<int> temp;
int d=0;

void dfs(int u,int dep)
{
	if(dep>d)
	{
		d=dep;
		res=temp;
	}
	
	for(auto j:g[u])
	{
		temp.push_back(j);
		dfs(j,dep+1);
		temp.pop_back(); 
	}
} 

int main()
{
	int n;cin>>n;
	for(int i=0;i<n;i++)
	{
		int k;cin>>k;
		for(int j=0;j<k;j++)
		{
			int x;cin>>x;
			st[x]=1;
			g[i].push_back(x);
		}
		sort(g[i].begin(),g[i].end());
	}
	
	int root=-1;
	for(int i=0;i<n;i++)
		if(!st[i])
		{
			root=i;
			dfs(root,0);
			break;
		}
	
	cout<<res.size()+1<<endl;
	cout<<root;
	for(auto x:res)
		cout<<" "<<x;
	
    return 0;
}