L2-036 网红点打卡攻略 (dfs)

分析

本题由于N较小，所以用邻接矩阵也没有问题。
把每个点输进去后要看一下是否能遍历到n个节点，能则进行dfs()，同时更新最小值。

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 210,M = N*N;
int h[N],e[M],w[M],ne[M],idx;
int n,m,pos[N],tot,ansco=INT_MAX,ansid,cnt;
void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int idx,int co,int J)
{
    if(u==0 && idx)
    {
        if(co<ansco)
        {
            ansco=co;
            ansid=J;
        }
        cnt++;
        return ;
    }
    for(int i=h[u];~i;i=ne[i])
    {
       int j=e[i];
       if(j==pos[idx])
       {
         dfs(j,idx+1,co+w[i],J);
       }
    }
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>m;
    int a,b,c;
    for(int i=0;i<m;i++)
    {
        cin>>a>>b>>c;
        add(a,b,c),add(b,a,c);  
    }
    int T; cin>>T;
    for(int j=1;j<=T;j++)
    {
        cin>>tot;
        set<int> s;
        bool flag=0;
        for(int i=0;i<tot;i++)
        {
            cin>>pos[i];
            if(s.count(pos[i])) flag=1; //一个点遍历了至少2次以上，方案不合法
            s.insert(pos[i]);
        }
        if(s.size()!=n) flag=1; //不能遍历到所有的网红点，非法
        pos[tot]=0;     //终点设置为0点
        if(flag){
            continue;   
        }
        else{
            dfs(0,0,0,j);
        }
    }
    cout<<cnt<<endl;
    cout<<ansid<<" "<<ansco<<endl;
    return 0;
}