nyoj115(城市平乱)（dijkstra算法）

#include<stdio.h>
#include<string.h>
#define INF 10000
int map[1007][1007];
int n,m,army[105];
int dijkstra(int q)
{
    int flag[1007],dis[1007];
    for(int i=1;i<=m;i++)
    {
        flag[i]=0;dis[i]=INF;
    }
    flag[q]=1;dis[q]=0;
    int s=m-1, min;
    while(s--)            //循环m-1次更新dis数组
    {
        for(int j=1;j<=m;j++)
        {
            if(map[q][j]&&!flag[i]&&dis[j]>map[q][j]+dis[q])
            dis[j]=map[q][j]+dis[q];
        }
        min=INF;
        for(int i=1;i<=m;i++)
        {
            if(dis[i]<min&&!flag[i])
            {
                min=dis[i];
                q=i;
            }
        }
        flag[q]=1;
    }
    min=INF;
    for(int i=1;i<=n;i++)
    {
        if(dis[army[i]]<min)
        min=dis[army[i]];
    }
    return min;
}
int main()
{
    int icase;
    scanf("%d",&icase);
    while(icase--)
    {
        memset(map,0,sizeof(map));
        int p,q;
        scanf("%d%d%d%d",&n,&m,&p,&q);
        for(int i=1;i<=n;i++)
        scanf("%d",&army[i]);
        int a,b,t;
        for(int i=1;i<=p;i++)
        {
            scanf("%d%d%d",&a,&b,&t);
            if(map[a][b])
            map[a][b]=map[b][a]=map[a][b]>t?t:map[a][b];
            else
            map[a][b]=map[b][a]=t;
        }
        printf("%d\n",dijkstra(q));
    }
    return 0;
}