【杭电oj】2066 - 一个人的旅行（SPFA）

题目链接

单源最短路，但是有多个终点，求到最近终点的距离。

思路：唯一的难点就是将草儿家看成0点，做一个炒鸡源点。

没注意多组输入WA1。

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int ,int> pii;
const int INF = 0x3f3f3f3f;
const int N = 5e3+10;
const int M = 5e3+10;

struct EDGE{
    int next;
    int to;
    int w;
}edge[M];

int n,m,d,cnt = 0;
int head[N],dis[N];
bool inq[N];

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

void SPFA(){
    memset(dis, INF, sizeof dis);
    memset(inq, 0, sizeof inq);
    queue<int> que; int p;
    que.push(0); dis[0] = 0; inq[0] = true;
    while(!que.empty()){
        p = que.front(); que.pop(); inq[p] = false;
        for(int i=head[p]; ~i; i=edge[i].next){
            int u = edge[i].to;
            if(dis[u] > dis[p] + edge[i].w){
                dis[u] = dis[p] + edge[i].w;
                if(!inq[u]){
                    inq[u] = true;
                    que.push(u);
                }
            }
        }
    }
    int ans = INF;
    for(int i=1; i<=d; ++i){
        scanf("%d", &p);
        ans = min(ans,dis[p]);
    }
    printf("%d\n", ans);
}
int main(int argc, char const *argv[])
{
    int u, v, w;
    while(scanf("%d %d %d",&n, &m, &d) != EOF){
        memset(head, -1, sizeof head);
        for(int i=1;i<=n;++i){
            scanf("%d %d %d",&u, &v, &w);
            add(u, v, w);
            add(v, u, w);
        }
        for(int i=1;i<=m;++i){
            scanf("%d",&u);
            add(u, 0, 0);
            add(0, u, 0);
        }
        SPFA();
    }
    return 0;
}