最短路+dfs【新年好】

题目链接

题目大意：给一个图 求其中6个点的全排列 哪种路最短
先预处理6个点到其他点的最短路
然后 dfs枚举排列 求最小值

#include<bits/stdc++.h>
using namespace std ;
const int N=1e6;
int dis[6][N],st[N];
int n,m,sorce[6];
int h[N],ne[N],e[N],w[N],idx;
void add(int a,int b,int c)
{
    e[idx]=b;
    w[idx]=c;
    ne[idx]=h[a];
    h[a]=idx++;
}
void spfa(int star,int *dis)
{
    memset(dis,0x3f,N*4);//这边是 N*4 不是sizeof dis
    dis[star]=0;
    queue<int>q;
    q.push(star);
    memset(st,0,sizeof st);
    st[star]=1;
    while(q.size())
    {
        int t=q.front();
        q.pop();
        st[t]=0;
        for(int i=h[t];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[t]+w[i])
            {
                dis[j]=dis[t]+w[i];
                if(st[j]==0)
                {
                    q.push(j);
                    st[j]=1;
                }
            }
        }
    }
}
int dfs(int u,int star,int distance)
{
    if(u>5) return distance;
    int res=0x3f3f3f3f;
    for(int i=1;i<=5;i++)
    {
       if(st[i]==0)
       {
           st[i]=1;
           res = min(res,dfs(u+1,i,distance+dis[star][sorce[i]]));
           st[i]=0;
       }
    }
    return res;
}

int main()
{
    cin>>n>>m;
    sorce[0]=1;
    memset(h,-1,sizeof h);
    for(int i=1;i<=5;i++) cin>>sorce[i];
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    for(int i=0;i<=5;i++) spfa(sorce[i],dis[i]);
    memset(st,0,sizeof st);
    cout<<dfs(1,0,0)<<endl;
}