dijkstra--求最短路--要判重边，否则wr

2019年3月17日 312次阅读 来源: Dijkstra算法

A – Til the Cows Come Home

POJ 2387：http://poj.org/problem?id=2387

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <algorithm>
using namespace std;
typedef pair<int, int> P;
//#define LOCAL
#define INF 0x3f3f3f3f
#define MAX_N 2005

int T, N;
int cost[MAX_N][MAX_N];
int d[MAX_N];
bool used[MAX_N];

void dijkstra(int s)//N*N的实践复杂度
{
    fill(d, d+N+1, INF);
    fill(used, used+N+1, false);
    d[s] = 0;
    while(true)
    {
        int v = -1;
        for(int u = 1; u <= N; u++) {
            if(!used[u] && (v == -1 || d[u] < d[v]))
            {
                v = u;//找到距离最小的点
            }
        }
        if(v == -1) break;
        used[v] = true;//标记这个点已经用过
        for(int u = 1; u <= N; u++) {
            d[u] = min(d[u], d[v] + cost[v][u]);
        }
    }
}

int main()
{
 #ifdef LOCAL
		freopen("b:\\data.in.txt", "r", stdin);
 #endif

    while(cin >> T >> N)
    {
        memset(cost, INF, sizeof(cost));
        for(int i = 0; i < T; i++)
        {
            int f, to, cos;
            cin >> f >> to >> cos;
            if(cost[f][to] > cos) {//判断重边的情况！！！！
                cost[f][to] = cos;
                cost[to][f] = cos;
            }

        }
        dijkstra(N);
        cout << d[1] << endl;
    }
    return 0;
}

    原文作者：Dijkstra算法
    原文地址: https://blog.csdn.net/AleiChen/article/details/51314038
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞