hdu-1874 畅通工程续(SPFA做法)

畅通工程续

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 37143    Accepted Submission(s): 13680

Problem Description

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

 

Input

本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

 

Output

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.

 

Sample Input

  
  

   
   3 3
0 1 1
0 2 3
1 2 1
0 2
3 1
0 1 1
1 2
  
  

 

Sample Output

  
  

   
   2
-1
  
  

 

# include<queue>
# include<stack>
# include<cstdio>
# include<cstring>
# include<algorithm>
# define MAXM  100000 + 100
# define INF 0x3f3f3f3f
using namespace std;
int n, m;
int st, en;
int next, cnt;
int head[MAXM], dist[MAXM];
int vis[MAXM];

struct node
{
	int from, to, val, next;
}p[MAXM];

void init()
{
	cnt = 0;
	memset(head, -1, sizeof(head));
	memset(vis, 0, sizeof(vis));
}

void add(int u, int v, int w)
{
	p[cnt].from = u;
	p[cnt].to = v;
	p[cnt].val = w;
	p[cnt].next = head[u];
	head[u] = cnt;
	cnt++;
}


void get()
{
	int a, b, c;
	while(m--)
	{
		scanf("%d%d%d",&a,&b,&c);
		add(a, b, c);
		add(b, a, c);
	}
}

void spfa(int ps)
{
	dist[ps] = 0;
	vis[ps] = 1;
	queue<int>Q;
	Q.push(ps);	
	while(!Q.empty())
	{
	    int u = Q.front();
		Q.pop();
		vis[u] = 0;
		
		for(int i = head[u]; i != -1; i = p[i].next)
		{
			int v = p[i].to;
			if(dist[v] > dist[u] + p[i].val)
			{
				dist[v] = dist[u] + p[i].val;
				if(!vis[v])
				{
					vis[v] = 1;
					Q.push(v);
				}
			}
		}
	}
	if(dist[en] != INF)
	printf("%d\n",dist[en]);
	else
	printf("-1\n");
}

int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i = 0; i < n; i++)
		 dist[i] = INF;
		init();		 
		get();	 
		scanf("%d%d",&st,&en);
		spfa(st);
	}
	return 0;
}