DIJ + DFS - CQOI 2005 新年好 - AcWing 1135

最新推荐文章于 2021-08-31 11:49:51 发布

njuptACMcxk

最新推荐文章于 2021-08-31 11:49:51 发布

阅读量183

点赞数

分类专栏：最短路文章标签：算法 dfs dijkstra 数据结构 acm竞赛

本文链接：https://blog.csdn.net/njuptACMcxk/article/details/106989827

版权

最短路专栏收录该内容

48 篇文章 0 订阅

订阅专栏

DIJ + DFS - CQOI 2005 新年好 - AcWing 1135

重庆城里有 n 个车站，m 条双向公路连接其中的某些车站。

每两个车站最多用一条公路连接，从任何一个车站出发都可以经过一条或者多条公路到达其他车站，但不同的路径需要花费的时间可能不同。

在一条路径上花费的时间等于路径上所有公路需要的时间之和。

佳佳的家在车站 1，他有五个亲戚，分别住在车站 a,b,c,d,e。

过年了，他需要从自己的家出发，拜访每个亲戚（顺序任意），给他们送去节日的祝福。

怎样走，才需要最少的时间？

输入格式

第一行：包含两个整数 n,m，分别表示车站数目和公路数目。

第二行：包含五个整数 a,b,c,d,e，分别表示五个亲戚所在车站编号。

以下 m 行，每行三个整数 x,y,t，表示公路连接的两个车站编号和时间。

输出格式

输出仅一行，包含一个整数 T，表示最少的总时间。

数据范围

$1≤n≤50000, 1≤m≤10^5, 1<a,b,c,d,e≤n, 1≤x,y≤n, 1≤t≤100$

输入样例：

输出样例：

分析：

$目标：从 1 号点出发，求一条经过 a, b, c, d, e 五个点的最短路。$

$要考虑 a, b, c, d, e 五个点经过的顺序，然后按照经过的顺序，在每个点两两间求最短路，累加得到总的最短距离。$

$因此我们先预处理以 1 号点, a, b, c, d, e ，六个点为源点的最短路径，$

$(预处理很重要，能够节约很多时间)$

$接着枚举 a, b, c, d, e 五个点经过的顺序，用全局变量 r e s 更新每种顺序下的最短距离即可。$

$注意：$

$s i z e o f 计算形式参数的数组大小不准确。$

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<iostream>
#include<vector>

#define P pair<int,int>

using namespace std;

const int N=50010, M=200010, inf=0x3f3f3f3f;

int n,m;
int source[6];
int e[M],ne[M],w[M],h[N],idx;
int dis[6][N];
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void dijkstra(int S,int dis[])
{
    memset(dis,0x3f,N*4);   //4字节
    //memset(dis,0x3f,sizeof dis);   //sizeof无法正确计算形式参数内的数组大小
    memset(st,false,sizeof st);
    dis[S]=0;
    
    priority_queue<P,vector<P>,greater<P> >heap;
    heap.push({0,S});
    
    while(heap.size())
    {
        P t=heap.top();
        heap.pop();
        
        int id=t.second;
        if(st[id]) continue;
        st[id]=true;
        
        for(int i=h[id];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[id]+w[i])
            {
                dis[j]=dis[id]+w[i];
                heap.push({dis[j],j});
            }
        }
    }  
}

int dfs(int u,int start,int distance)
{
    if(u==6) return distance;
    
    int res=inf;
    for(int i=1;i<=5;i++)
        if(!st[i])
        {
            int next=source[i];
            st[i]=true;
            res=min(res,dfs(u+1,i,distance + dis[start][next]));
            st[i]=false;
        }
        
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    source[0]=1;
    for(int i=1;i<=5;i++) scanf("%d",&source[i]);
    
    memset(h,-1,sizeof h);
    int a,b,c;
    while(m--)
    {
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    
    for(int i=0;i<6;i++) dijkstra(source[i],dis[i]);
    
    memset(st,false,sizeof st);  //清空dijkstra标记过的st数组
    printf("%d\n",dfs(1,0,0));

    return 0;
    
}