蓝桥杯练习题 最短距离

问题描述

给定一个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<iostream>
#include<string>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 20005;
const int inf = 1<<30;
int n,m,pos;
struct node
{
	int to,w,next;
}edge[maxn*15];
int head[maxn];
int dis[maxn];
bool vis[maxn];
void add( int u,int v,int w )
{
	edge[++pos].to = v;  edge[pos].w = w;
	edge[pos].next = head[u]; head[u] = pos;
}
typedef pair<int,int>pii; 
void Dijkstra( int s )
{
	for( int i = 1; i <= n; i ++ )
		dis[i] = inf;
	memset( vis,0,sizeof(vis) );
	priority_queue<pii,vector<pii>,greater<pii> >que;  
    dis[s] = 0;  
    que.push( make_pair(dis[s],s) );  
	while( !que.empty() ){
		pii u = que.top(); que.pop();
		int x = u.second;
		if( vis[x] )	continue;
		vis[x] = true;
		for( int i = head[x]; i != -1; i = edge[i].next ){
			node v = edge[i];
			if( dis[v.to] > dis[x] + v.w ){
				dis[v.to] = dis[x] + v.w;
				que.push( make_pair(dis[v.to],v.to) );
			}
		}
	}
}
int main()
{
	//freopen("data.txt","r",stdin);
	int u,v,w;
	pos = 0;
	scanf("%d%d",&n,&m);
	memset(head,-1,sizeof(head));
	for( int i = 1; i <= m; i ++ ){
		scanf("%d%d%d",&u,&v,&w);
		add( u,v,w );
	}
	Dijkstra(1);
	for( int i = 2; i <= n; i ++ )
		printf("%d\n",dis[i]);
	return 0;
}

#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 20005;
const int inf = 1<<30;
int n,m,pos;
struct node
{
	int to,w,next;
}edge[maxn*15];
int head[maxn];
int dis[maxn];
bool vis[maxn];
void add( int u,int v,int w )
{
	edge[++pos].to = v;  edge[pos].w = w;
	edge[pos].next = head[u]; head[u] = pos;
}
void SPEA( int s )
{
	queue<int>que;
	bool inq[maxn] = {0};
	for( int i = 1; i <= n; i ++ )
		dis[i] = inf;
	dis[s] = 0;
	que.push(s);
	while( !que.empty() )
	{
		int x = que.front(); que.pop();
		inq[x] = false;
		for( int i = head[x]; i != -1; i = edge[i].next ){
			node &v = edge[i];
			if( dis[v.to] > dis[x] + v.w ){
				dis[v.to] = dis[x] + v.w;
				if( !inq[v.to] ){
					inq[v.to] = true;
					que.push(v.to);
				}
			}
		}
	}
}
int main()
{
	//freopen("data.txt","r",stdin);
	int u,v,w;
	pos = 0;
	scanf("%d%d",&n,&m);
	memset(head,-1,sizeof(head));
	for( int i = 1; i <= m; i ++ ){
		scanf("%d%d%d",&u,&v,&w);
		add( u,v,w );
	}
	SPEA(1);
	for( int i = 2; i <= n; i ++ )
		printf("%d\n",dis[i]);
	return 0;
}