1135. 新年好

题目大意

题目链接

AC模板

• 暴力枚举所有的走访顺序，假设顺序5！*O(m)5k,k为是spfa的常数,可以考虑先求五个亲戚和家到其他站点的最短路，
• 枚举拜访顺序的时候直接查表即可,这样的复杂度为O(m)*5+5!,思路类似于预处理然后查表

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<queue>
using namespace std;
typedef pair<int,int> PII;
const int N=50010,M=2e5+10,INF=0x3f3f3f3f;
int n,m;
int h[N],e[M],ne[M],w[M],idx;
int dist[10][N];

bool st[N];
int s[10];
int res=INF;
void init()
{
    memset(h,-1,sizeof(h));
    idx=0;
}
void add(int a,int b,int c)
{
    e[idx]=b;
    w[idx]=c;
    ne[idx]=h[a];
    h[a]=idx++;
}
void dijkstra(int start,int dist[])
{
    memset(st,0,sizeof(st));
    memset(dist,0x3f,N*4);
    priority_queue<PII,vector<PII>,greater<PII> >q;
    dist[start]=0;
    q.push({0,start});
    while(q.size())
    {
        auto t=q.top();
        q.pop();
        int d=t.first;
        int ver=t.second;
        if(st[ver]) continue;
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[ver]+w[i])
            {
                dist[j]=dist[ver]+w[i];
                q.push({dist[j],j});
            }
        }
    }
}
 //枚举摆放顺序,一共5个空格，正在填第1个，当前的起点是start,目前花费的时间为0
void  dfs(int u,int start,int distance)
{
    if(u==6) 
    {
        res=min(res,distance);
        return;
    }
    for(int i=1;i<=5;i++)
    {
        if(!st[i])
      {
        int next=s[i];
        st[i]=true;
        dfs(u+1,i,distance+dist[start][next]);
        st[i]=false;
      }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    s[0]=1;
    for(int i=1;i<=5;i++)
    scanf("%d",&s[i]);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    for(int i=0;i<6;i++)
    dijkstra(s[i],dist[i]);
    memset(st,0,sizeof(st));
    dfs(1,0,0);
    cout<<res<<endl;
    return 0;
}