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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值