首页 > 代码库 > 图论——最短路径——dij

图论——最短路径——dij

dij 模板

通过 poj 2387 求去点1 -点 n 的最短路径

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>

using namespace std;

const int MAXV = 4010;
const int inf = 10000000;

int map[MAXV][MAXV];//路径长度
int d[MAXV];//距离
bool vis[MAXV];
int n,m;

void dijkstra(int s)
{
	for(int i=1;i<=n;i++)
    {
		vis[i]=0;
		d[i]=map[s][i];
	}

	while (1)
    {
		int min=inf,v = -1;
		for(int i=1;i<=n;i++)
			if(!vis[i] && d[i]<min)
			{
				v=i;
				min=d[i];
			}
        if(v == -1)
            break;
		vis[v]=1;
		for(int i=1;i<=n;i++)
			if(!vis[i] && d[i] > d[v] + map[v][i])
				d[i]=map[v][i]+d[v];
	}
}

int main()
{
	int i,j,a,b,c;
	while(scanf("%d%d",&m,&n) != EOF)
    {
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
            {
                if(i==j)
                    map[i][i]=0;
                else
                    map[i][j]=map[j][i]=inf;
            }
            for(i=1;i<=m;i++)
            {
                scanf("%d%d%d",&a,&b,&c);
                if(map[a][b]>c)
                    map[a][b]=map[b][a]=c;
            }
		dijkstra(1);
		printf("%d\n",d[n]);
	}
	return 0;
}


图论——最短路径——dij