HDU 2680 Choose the best route

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=2680

题意：给你n,m,s三个数，n是公交站的数目并且编号分别为1，2……….n.
m是是公交车站之间可行的路，代表后面会有m行数，每行3个数p,q,t（表示从p点到q点需要花费时间t，要注意的是这是一个单向图，因为题目说means from station p to station q）。s是Kiki要去的目的地编号。然后又给了一个w，接下来w个数，代表接下来w个数是可以作为起点的编号，最后求Kiki到目的地所花费的最少时间。
思路：因为按照题目所说起点可有多个，我们可以虚拟出一个节点（不妨虚拟出一个编号为0的节点），是Kiki最初的位置，再把Kiki可到达的几个车站与其距离初始化为0即可按照普通的迪克斯特拉算法写。

AC代码：

#include<stdio.h>
#include<string.h>
#include<string>
#include<iostream>
using namespace std;
#define INF 0x3f3f3f3f
int map[1005][1005];
int flag[1005],way[1005];
void dij(int n,int s)
{
    for(int i=0; i<=n; i++)
    {
        way[i]=map[s][i];
    }
    for(int i=0; i<n; i++)
    {
        int min=INF;
        int k=0;
        for(int j=0; j<=n; j++)
        {
            if(flag[j]==0&&way[j]<min)
            {
                min=way[j];
                k=j;
            }
        }
        flag[k]=1;
        if(k==s)
            break;
        for(int j=0; j<=n; j++) //找出距离刚刚选出来的点最近的距离组成新的路径
        {
            if(flag[j]==0&&way[k]+map[k][j]<way[j])
            {
                way[j]=way[k]+map[k][j];
            }
        }
    }
}
int main()
{
    int n,m,s;
    while(~scanf("%d %d %d",&n,&m,&s))
    {
        for(int i=0;i<1002;i++)
            for(int j=0;j<1002;j++)
        {
            if(i!=j)
                map[i][j]=INF;
            else
                map[i][j]=0;
        }
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            if(map[u][v]>w)
            map[u][v]=w;//有向图
        }
        int w,d;
        scanf("%d",&w);
        while(w--)
        {//把Kiki可到达的几个车站与节点为0的距离初始化为0
            scanf("%d",&d);
            map[0][d]=0;
        }
        memset(flag,0,sizeof(flag));
        flag[0]=1;
        //将编号为0的节点虚拟为Kiki的起点
        dij(n,0);
        if(way[s]!=INF)
            printf("%d\n",way[s]);
        else
            printf("-1\n");
    }
    return 0;
}