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

猜你喜欢

转载自blog.csdn.net/qq_45550375/article/details/123540110