[HDU 2544] 最短路 spfa写法

spfa是对迪杰斯特拉算法的队列优化,套路大概就是:

让dis[1] 入队, 然后判断队列是否为空;

让它赋给队首, 然后就是一样的循环;

唯一的区别就是这个地方!vis[i] 是拆开放在后面的;


AC代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <queue>
#define ll long long
#define inf 9999999
using namespace std ;
int dis[2000] , vis[2000] , map[200][2000] , u , v ;
void input();
void init();
void spfa();
void output();

int main()
{
	while(scanf("%d%d",&v,&u),v+u)
	{
		init();
		input();
		spfa();
		output();
	}
}

void input()
{
	for(int i = 0 ; i < u;i++)
	{
		int a , b , len ; 
		scanf("%d%d%d",&a,&b,&len);
		if(map[a][b] > len)
		{
			map[a][b] = map[b][a] = len ;
		}
	}
}

void init()
{
	for(int i = 0 ; i <=v ;i++)
	{
		for(int j = 0 ; j<=v;j++)
		{
			map[i][j] = inf ;
		}
		dis[i] = inf ;
		vis[i] = 0 ;
		map[i][i] = 0 ;
	}
}

void output()
{
	printf("%d\n",dis[v]);
}

void spfa()
{
	queue<int>Q;
	dis[1] = 0 ;
	Q.push(1);
	vis[1] = 1 ;//*初始点肯定是会被使用了的
	while(!Q.empty())
	{
		int now = Q.front();
		Q.pop();
		vis[now] = 0 ;  //*这个地方是0不是1 ; 因为出队时候这个now还并没有被使用
		for(int i = 1 ; i <=v ;i++)
		{
			if(dis[i] > dis[now] + map[now][i])
			{
				dis[i] = dis[now] + map[now][i];
				if(!vis[i])
				{
					Q.push(i);
					vis[i] = 1 ;//*使用了后就标记
				}
			}
		}
	}
}



你可能感兴趣的:([HDU 2544] 最短路 spfa写法)