最短路径快速spfa

#include <stdio.h>
#include <queue>
#include <iostream>
using namespace std;
#define INF 0xfffff					//因为为了辨别是否有负权，所以INF不能开太大
#define MAX 1100
int dist[MAX], pre[MAX], path[MAX][MAX];
bool sign[MAX];
void initialize(int n)			//初始化
{
	for(int i=1; i<=n; i++)
	{
		{
			//pre[i] = 0;
			dist[i] = INF;		//将距离开始全变为最大
			sign[i] = false;
		}
		for(int j=1; j<=n; j++)
		path[i][j] = INF;		//图初始
	}
}
void spfa(int n, int start)							//无法计算负权
{
   /* for (int i=1; i<=n; ++i)                      //初始化
    {
        dist[i] = INF;
        sign[i] = false;
    }*/
    queue<int> Q;
    dist[start] = 0;
    sign[start] = true;
    Q.push(start);
    while (!Q.empty()){
        int temp = Q.front();
        Q.pop();
        for (int i=0; i<=n; ++i)
        {
            if (dist[temp] + path[temp][i] < dist[i])//存在负权的话，就需要创建一个COUNT数组，当某点的入队次数超过V(顶点数)返回。
            {
                dist[i] = dist[temp] + path[temp][i];
                if (!sign[i])
                {
                    Q.push(i);
                    sign[i] = true;
                }
				//这个内循环可以判断所要权值相对应条件的值如dist[start];
            }
        }
        sign[temp] = false;
    }
}
void input(int line)
{
	int a, b, weight;
	for(int i=0; i<line; i++)
	{
		scanf("%d%d%d", &a, &b, &weight);
		if(path[a][b] > weight)				//有多条路，保存最短的那条
		{
			path[a][b] = weight;
			path[b][a] = weight;			//无向图双向
		}
	}
}
int main()
{
	int n, line;
	scanf("%d%d", &n, &line);
	initialize(n);
	input(line);
	spfa(n, 1);
	printf("%d\n\n", dist[n]);
	return 0;
}

来自为知笔记(Wiz)

附件列表