L2-038 病毒溯源 (dfs，24与25分)

L2-038 病毒溯源 (25 分)

思路1（24分）：

vector<int> v[N]建立邻接表，之后对v[i]从大到小排序，便于寻找最小序列，但是只得了24分

代码（24分）：

#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define f(i,a,n) for(int i=a;i<n;++i)
#define ff(i,a,n) for(int i=a;i<=n;++i)
const int INF=0x3f3f3f3f;
using namespace std;
typedef long long ll;
//
const int N=1e4+5;
int n,st[N],dep[N],pr[N],start;
vector<int> v[N];

int mx_dep,p;
void dfs(int rt,int d){
    if(!v[rt].size()){
        if(dep[rt]==mx_dep)p=rt;
        return;
    }
    for(int i=0;i<v[rt].size();i++){
    	//错误示范：
        //pr[v[rt][i]]=i;
        pr[v[rt][i]]=rt;
        dep[v[rt][i]]=d+1;
        mx_dep=max(mx_dep,d+1);
        dfs(v[rt][i],d+1);
    }
}
void pri(int x){
    if(x!=start)pri(pr[x]);
    if(x!=p)cout<<x<<" ";
	else cout<<x;
}
int main(){
    cin>>n;
    f(i,0,n){
        int t;cin>>t;
        f(j,0,t){
            int k;cin>>k;
            v[i].push_back(k);
            st[k]=1;
        }
        //从大到小排序，便于寻找最小序列
        sort(v[i].begin(),v[i].end());
        reverse(v[i].begin(),v[i].end());
    }
    f(i,0,n){
        if(!st[i]){
            start=i;
            break;
        }
    }
    dfs(start, 1);
    cout<<mx_dep<<endl;
    pri(p);
    return 0;
}

思路2（25分）：

在dfs过程中回溯，更新最短路，利用vecotr 元素天然可比较性，较为方便，在此用链式前向星建图

代码（25分）：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//
const int N = 1e4 + 10,M=1000010;
int h[N],e[M],ne[M],idx;
int n;
int mx_dep;
bool st[N];
vector<int> temp,res;

void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u,int dep){
    if(dep > mx_dep){
        mx_dep=dep;
        res=temp;
    }
    else if(dep == mx_dep && temp < res){
        res=temp;
    }
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        temp.push_back(j);
        dfs(j, dep + 1);
        //树形结构，方便回溯
        temp.pop_back();
    }
}

int main(){
    memset(h,-1,sizeof h);
    cin>>n;
    for(int i=0;i<n;i++){
        int k;cin>>k;
        while(k--){
            int a;cin>>a;
            add(i,a);
            st[a]=true;
        }
    }
    int root=0;
    while(st[root])root++;
    dfs(root,1);
    cout << mx_dep << endl;
    cout<<root;
    for(int i=0; i < res.size(); i++){
        cout << ' ' << res[i];
    }
    return 0;
}