HDU 2066(dijkstra算法模板题)

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

标准最短路径问题，起点可以看作是草儿家，从草儿家到与家相连的地方time=0，dijkstra求最短路径，然后找出想去地方的最小代价就行了，经典模板题，但是WA了好几次，开始找不出错误，后来发现体重有句话：(1=<(a,b)<=1000;a,b 之间可能有多条路），瞬时明白了，两点之间可以有多条路径的。 

 

当然我的代码中，dijkstra 中的 n有些大了，我们可以在主函数中记录下u,v的最大值作为n就够了，不必每次都遍历完

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define INF 100000000
#define n 1000

using namespace std;
int map[1100][1100],dis[1100],flag[1100];
int T,S,D;
void dijkstra()
{
    int i,j,k,min;
    memset(flag,0,sizeof(flag));
    flag[0]=1;
    for(i=0;i<=n;i++)
    {
        dis[i]=map[0][i];
    }
    for(i=1;i<n;i++)
    {
        min=INF;
        for(j=1;j<=n;j++)
        {
            if(!flag[j]&&dis[j]<min)
            {
                min=dis[j];
                k=j;
            }
        }
        flag[k]=1;
        for(j=1;j<=n;j++)
        {
            if(!flag[j]&&dis[j]>dis[k]+map[k][j])
                dis[j]=dis[k]+map[k][j];
        }
    }
}
int main()
{
    int u,v,w,x,y,min;
    while(scanf("%d%d%d",&T,&S,&D)!=EOF)
    {
        for(int i=0;i<1010;i++)
            for(int j=0;j<1010;j++)
                map[i][j]=INF;
        for(int i=1;i<=T;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            map[u][v]=map[u][v]>w?w:map[u][v];   //可能两点之间有多条路径
            map[v][u]=map[u][v];
        }
        for(int i=1;i<=S;i++)
        {
            scanf("%d",&x);
            map[0][x]=0;
            map[x][0]=0;
        }
        dijkstra();
        min=INF;
        for(int i=0;i<D;i++)
        {
            scanf("%d",&y);
            if(min>dis[y]) min=dis[y];
        }
        printf("%d\n",min);
    }
    return 0;
}