Vijos1767-YYB喋血

描述

---

Hwz把YYB放到了一个迷宫之中，这个迷宫由n个节点构成，两个节点之间可能存在多条无向边，YYB的起点为1号节点，终点为n号节点。有m条无向边，对于每一条无向边，存在一个喋血值（∈N*，且≤100），即走过这条边的花费。另外，还有k个节点上有治疗药，即若YYB走到这个节点上时(不妨称这个点为治愈点)，他身上所累积的喋血值会归零。YYB希望以最小的喋血值走完迷宫。

输入格式

---

第1行n,m,k分别表示有n个节点，m条无向边，以及k个治愈点。
第2行到m+1行 每一行有一个x,y,z表示x到y有一条喋血值为z的无向边
第n+2行 有k个整数，分别为治愈点的号数
PS：保证数据中没有负权回路。保证治愈点不重复。

输出格式

---

一行minblood 表示YYB走完迷宫的最小喋血值
当然，如果无法走出迷宫，输出Oh no!

Input
3 3 1
1 2 100
2 3 1
1 3 3
2

-

Output
1

Hint

---

范围：
对于100%的数据
1≤n≤5000,1≤k≤n,1≤m≤25000

-

题解

---

用并查集判断“回血点”与起点的连通性，若与起点连通则加一条边权为0的边连接回血点和起点。建好图后跑一遍SPFA求出最短路径。

代码

---

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;

const int MAXN=5000+10;
const int MAXM=25000+10;

struct edge{int to,next,v;}e[MAXM*2];
int n,m,k,cnt,head[MAXN],fa[MAXN],dis[MAXN];
bool vis[MAXN];

int find(int x){return x==fa[x] ? x:fa[x]=find(fa[x]);}

inline void add(int _u,int _v,int _w)
{
    cnt++;
    e[cnt].to=_v;
    e[cnt].v=_w;
    e[cnt].next=head[_u];
    head[_u]=cnt;
}

void spfa(int s)
{
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    queue<int>q;
    while(!q.empty()) q.pop();
    q.push(s);
    dis[s]=0;
    vis[s]=true;
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        vis[t]=false;
        for(int i=head[t];i;i=e[i].next)
        {
            int temp=e[i].to;
            if(dis[temp]>dis[t]+e[i].v)
            {
                dis[temp]=dis[t]+e[i].v;
                if(!vis[temp])
                {
                    vis[temp]=true;
                    q.push(temp);
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++) 
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);add(v,u,w);
        if(find(u)!=find(v)) fa[fa[u]]=v;
    }
    if(find(1)!=find(n)) {printf("Oh no!\n");return 0;}
    int kx;
    for(int i=1;i<=k;i++)
    {
        scanf("%d",&kx);
        if(find(kx)==find(1)) add(1,kx,0);
    }
    spfa(1);
    printf("%d\n",dis[n]);
    return 0;
}