short-path problem (Dijkstra)

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;
const int INF = 0x3fffffff;

int  g[1005][1005];
int m;

int Dijkstra(int s,int t)
{
    bool visit[1005];
    int dis[1005];
    for(int i = 1; i <= m; ++i)
    {
        visit[i] = false;
        dis[i] = g[s][i];
    }
    visit[s] = true;

    int min = INF;
    int k;
    for(int i = 1; i < m; ++i)
    {
        min = INF;
        k = 1;
        for(int j = 1; j <= m; ++j)
        {
            if(!visit[j] && dis[j] < min)
            {
                min = dis[j];
                k = j;
            }
        }
        visit[k] = true;
        for(int j = 1; j <= m; ++j)
        {
            if(!visit[j] && dis[k] + g[k][j] < dis[j])
            {
                dis[j] = dis[k] + g[k][j];
            }
        }
    }
    return dis[t];
}

int main()
{
    int T, a, b, len;
    int s, t, p;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &m, &p);
        scanf("%d%d",&s,&t);
        for(int i = 1; i <= m; ++i)
            for(int j = 1; j < i; ++j)
                g[i][j] = g[j][i] = INF;

        while(p--)
        {
            scanf("%d%d%d", &a, &b, &len);
            if(g[a][b] > len)
                g[a][b] = g[b][a] = len;
        }

        printf("%d\n", Dijkstra(s,t));
    }
    return 0;
}

Code From Nyoj 115


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值