CJOI 05新年好 (最短路+枚举)

Posted 2020-08-29 YJSheep

CJOI 05新年好 (最短路+枚举)

重庆城里有n个车站，m条双向公路连接其中的某些车站。每两个车站最多用一条公路连接，从任何一个车站出发都可以经过一条或者多条公路到达其他车站，但不同的路径需要花费的时间可能不同。在一条路径上花费的时间等于路径上所有公路需要的时间之和。

佳佳的家在车站1，他有五个亲戚，分别住在车站a,b,c,d,e。过年了，他需要从自己的家出发，拜访每个亲戚（顺序任意），给他们送去节日的祝福。怎样走，才需要最少的时间？

输入

　第一行：n(n<=50,000),m(m<=100,000)为车站数目和公路的数目。

第二行：a,b,c,d,e，为五个亲戚所在车站编号（1<a,b,c,d,e<=n）。

以下m行，每行三个整数x,y,t（1<=x,y<=n,1<=t<=100）,为公路连接的两个车站编号和时间。

输出

　仅一行，包含一个整数T，为最少的总时间

样例输入

6  6

2  3  4  5  6

1  2  8

2  3  3

3  4  4

4  5  5

5  6  2

1  6  7

样例输出

21

 

解题报告

必须经过a,b,c,d,e这几个点，故我们必须知道包括起点在内的6个点的最短路。所以跑6遍堆优dijkstra。最后再对a,b,c,d,e的顺序进行全排列，计算结果取最小即可。

复杂度O(6*nlogn)

 

#include<bits/stdc++.h>
#define Pair pair<int,int>
#define MAXN 70000+10
#define MAXM 200000+10
using namespace std;
struct Edge{
    int next,to,dis;
}edge[MAXM];
int n,m,num,v[MAXN],dis[MAXN],head[MAXN];
int disi[9][MAXN],u[MAXN],ans=99999999,a[8],b[8];
int temp=0;
void add(int from,int to,int dis)
{
    edge[++num].next=head[from];
    edge[num].to=to;
    edge[num].dis=dis;
    head[from]=num;
}
void dij(int s)
{
    memset(v,0,sizeof(v));
    priority_queue<Pair,vector<Pair>,greater<Pair> > h;
    for(int i=1;i<=n;i++) disi[s][i]=99999999;
    disi[s][u[s]]=0;
    h.push(Pair(0,u[s]));
    while(h.size()>0)
    {
        int k=h.top().second;h.pop();
        if(v[k]) continue;
        v[k]=1;
        for(int i=head[k];i;i=edge[i].next)
        if(disi[s][edge[i].to]>disi[s][k]+edge[i].dis)
        {
            disi[s][edge[i].to]=disi[s][k]+edge[i].dis;
            h.push(Pair(disi[s][edge[i].to],edge[i].to));
        }
    }
}
void dfs(int x)
{
    for(int i=1;i<=5;i++)
    if(!b[i])
    {
        b[i]=1;
        a[x]=i;
        if(x==5)
        {
            temp=0;
            for(int i=0;i<5;i++)
                temp+=disi[a[i]][u[a[i+1]]];
            ans=min(ans,temp);
            temp=0;
        }else dfs(x+1);
        b[i]=0;
        a[x]=0;
    }    
}
int main()
{
    freopen("newyear.in","r",stdin);
    freopen("newyear.out","w",stdout);

    scanf("%d%d",&n,&m);
    for(int i=1;i<=5;i++)
        scanf("%d",&u[i]);
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    
    u[0]=1;
    for(int i=0;i<=5;i++)
        dij(i);
    a[0]=0;
    dfs(1);
    printf("%d\n",ans);
    return 0;
}