【NEFU 1283】最短路

最新推荐文章于 2023-08-01 19:54:22 发布

原创最新推荐文章于 2023-08-01 19:54:22 发布 · 279 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

本文详细解析了在ACM竞赛中解决最短路径问题的算法实现，通过两次运行最短路径算法，分别计算从起点到各旅游点及从各旅游点返回起点的最短距离，适用于单向路径问题。

题目链接：http://acm.nefu.edu.cn/JudgeOnline/problemShow.php?problem_id=1283

这道题很简单，就是跑两遍最短路，因为是单向路径，所以跑一遍最短路算出到各个要旅游的地方的最短距离，然后就要算各个旅游的地方到家的最短距离，这里要把原来的路线反过来再跑一遍最短路，答案就出来了

代码：

#include <map>
#include <string>
#include <vector>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;

const int maxn=50005;

struct node
{
    int next,to,w;
}e[maxn];

queue<int>q;
int dis[maxn],vis[maxn];
int head[maxn],cnt;
int a[maxn],b[maxn],c[maxn],x[maxn];

void add(int u,int v,int w)
{
    e[++cnt].next=head[u];
    e[cnt].w=w;
    e[cnt].to=v;
    head[u]=cnt;
}

void spfa(int x)
{
    while(!q.empty())
    q.pop();
    dis[x]=0;
    q.push(x);
    vis[x]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u];i!=0;i=e[i].next)
        {
            node v=e[i];
            if(dis[v.to]>dis[u]+v.w)
            {
                dis[v.to]=dis[u]+v.w;
                if(!vis[v.to])
                {
                    q.push(v.to);
                    vis[v.to]=1;
                }
            }
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    for(int j=1;j<=t;j++)
    {
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        memset(head,0,sizeof(head));
        cnt=0;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a[i],&b[i],&c[i]);
            add(a[i],b[i],c[i]);
        }
        for(int i=1;i<=n;i++)
        {
            dis[i]=INF;
        }
        spfa(1);
        long long ans=0;
        int flag=0;
        for(int i=1;i<=k;i++)
        {
            scanf("%d",&x[i]);
            if(dis[x[i]]!=INF)
            {
                ans+=dis[x[i]];
            }
            else
            {
                flag=1;
            }
        }
        memset(head,0,sizeof(head));
        cnt=0;
        for(int i=1;i<=m;i++)
        {
            add(b[i],a[i],c[i]);
        }
        for(int i=1;i<=n;i++)
        {
            dis[i]=INF;
        }
        spfa(1);
        for(int i=1;i<=k;i++)
        {
            if(dis[x[i]]!=INF)
            {
                ans+=dis[x[i]];
            }
            else
            {
                flag=1;
            }
        }
        if(flag==1)
        printf("Case #%d: QAQ\n",j);
        else
        printf("Case #%d: %lld\n",j,ans);
    }
    return 0;
}