最短路【SPFA】

問題描述

給定一個n個頂點，m條邊的有向圖（其中某些邊權可能爲負，但保證沒有負環）。請你計算從1號點到其他點的最短路（頂點從1到n編號）。

輸入格式

第一行兩個整數n, m。

接下來的m行，每行有三個整數u, v, l，表示u到v有一條長度爲l的邊。

輸出格式

共n-1行，第i行表示1號點到i+1號點的最短路。

樣例輸入

3 3
1 2 -1
2 3 -1
3 1 2

樣例輸出

-1
-2

數據規模與約定

對於10%的數據，n = 2，m = 2。

對於30%的數據，n <= 5，m <= 10。

對於100%的數據，1 <= n <= 20000，1 <= m <= 200000，-10000 <= l <= 10000，保證從任意頂點都能到達其他所有頂點。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int next[200010] , f[200010], vis[20010], n, m, dis[20010], que[200010];
struct point
{
	int u, v, w;
}p[200010];
int min(int x, int y)
{
	return x<y?x:y;
}
void spfa()
{
	memset(vis, 0, sizeof(vis));
	memset(dis, 0, sizeof(dis));
	vis[1] = 1;
	for(int i = 2; i <= n; ++i)
	dis[i] = INF;
	int front = 0, rear = 0;
	que[rear++] = 1;
	while(front < rear)
	{
		int u = que[front++];
		vis[u] = 0;
		for(int i = f[u]; i != -1; i = next[i])
		{
			int v = p[i].v;
			if(dis[v] > dis[u]+p[i].w)
			{
				dis[v] = dis[u]+p[i].w;
				if(!vis[v])
				{
					que[rear++] = v;
					vis[v] = 1;
				}
			}
		} 
	} 
}
int main()
{
	int i, j, k;
	while(scanf("%d%d", &n, &m) != EOF)
	{
		memset(f, -1, sizeof(f));
		memset(next, -1, sizeof(next));
		for(i = 0; i < m; ++i)
		{
			scanf("%d%d%d", &p[i].u, &p[i].v, &p[i].w);
			next[i] = f[p[i].u];
			f[p[i].u] = i;
		}
		spfa();
		for(i = 2; i <= n; ++i)
		printf("%d\n", dis[i]); 
	}
	return 0;
} 

題意：就是給一個圖求最短路。

思路：用djs+鄰接矩陣超時了，所以用了spfa+鄰接表。