HDU2544,dijkstra算法

模板题.

/*************************************************************************
    > File Name: main.cpp
    > Author:Eagles 
    > Mail:None 
    > Created Time: 2018年09月05日 星期三 14时42分07秒
    > Description:HDU2544,dijkstra
 ************************************************************************/

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
#define N 200
int Gra[N][N];
bool vis[N];
int dis[N];
int n,m;
const int INF=1253452;
void init()
{
    memset(Gra,0x3f3f3f3f,sizeof(Gra));
    memset(vis,false,sizeof(vis));

    for (int i=0; i<m; i++)
    {
        int a,b,val;
        scanf("%d%d%d",&a,&b,&val);
        if (val<Gra[a-1][b-1])
        Gra[a-1][b-1]=Gra[b-1][a-1]=val;
    }

    for (int i=0; i<n; i++)
        dis[i]=Gra[0][i];

    vis[0]=true;
}

void dijkstra()
{
    for (int i=0; i<n-1; i++)
    {
        int minn=INF;
        int tmp;

        for (int j=0; j<n;j++)
        {
            if (!vis[j]&&dis[j]<minn)
            {
                minn=dis[j];
                tmp=j;
            }
        }

        vis[tmp]=true;

        for (int j=0; j<n; j++)
        {
            if (!vis[j]&&(dis[j]>(dis[tmp]+Gra[tmp][j])))
                dis[j]=dis[tmp]+Gra[tmp][j];
        }
    }

    printf("%d\n",dis[n-1]);
}
void print()
{
    for (int i=0; i<n ;i++)
    {
        for (int j=0; j<n; j++)
            printf("%d ",Gra[i][j]);
        printf("\n");
    }

    for (int i=0; i<n; i++)
        printf("%d ",dis[i]);
    printf("\n");
}
int main()
{
    while (~scanf("%d%d",&n,&m)&&(m||n))
    {
        init();
        //print();
        dijkstra();
    }
    return 0;
}
发布了45 篇原创文章 · 获赞 2 · 访问量 3099

猜你喜欢

转载自blog.csdn.net/wysiwygo/article/details/82427983
今日推荐