蓝桥杯-算法锻炼-最短路
问题描述
给定一个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,保证从任意顶点都能到达其他所有顶点。
什么鬼,开始没认真看数据范围,以为都是两万之内,原来m小于二十万,所以超时了,只给了七十分,这两天看了四种关最短路径的问题,diijkstra,flody,bellman-ford,SPFA,然并卵,没有真正掌握它们,但是我还是有一些总结的,关于这四种算法:
dijkstra适用于边权没负数,且数据可处理大一些的,二而flody代码简单,但容易超时,也是处理无负数边,这两个都不能处理回向图;而bellman可处理负数边权,但是数据大了会超时,用SPFA可处理超时问题,但是它用了队列知识,并且代码较其它复杂。
这道题要用SPFA算法,bellman算法会超时
定理: 只要最短路径存在,上述SPFA算法必定能求出最小值。
证明:
每次将点放入队尾,都是经过松弛操作达到的。换言之,每次的优化将会有某个点v的最短路径估计值d[v]变小。所以算法的执行会使d越来越小。由于我们假定图中不存在负权回路,所以每个结点都有最短路径值。因此,算法不会无限执行下去,随着d值的逐渐变小,直到到达最短路径值时,算法结束,这时的最短路径估计值就是对应结点的最短路径值。(证毕)
期望的时间复杂度O(ke), 其中k为所有顶点进队的平均次数,可以证明k一般小于等于2
#include <iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
#define MAX 200001
#define INF 99999999
typedef struct XNode
{
int pow;//边的权值
int adjvex;//点
struct XNode *next;
}Node;
Node * head[MAX];
int visit[MAX],dis[MAX];
void add(int u,int v,int w)
{
Node *p;
p=(Node*)malloc(sizeof(Node));
p->pow=w;
p->adjvex=v;
p->next=head[u];
head[u]=p;
}
void SPFA(int n)
{
int u,v,w;
queue<int>Q;//这个算法关键使用队列来处理问题
Node *p;
for(int i=1;i<=n;i++)//初始化
{
visit[i]=0;
dis[i]=INF;
}
Q.push(1);//头结点进队
dis[1]=0;
visit[1]=1;
while(!Q.empty())//直到队列为空才结束循环
{
u=Q.front();//头结点
Q.pop();//进队
for(p=head[u];p!=NULL;p=p->next)
{
v=p->adjvex;
w=p->pow;
if(dis[u]+w<dis[v])
{
dis[v]=dis[u]+w;
if(!visit[v])
{
visit[v]=1;
Q.push(v);//出队
}
}
}
}
}
int main()
{
int n,m,u,v,w;
while(cin>>n>>m){
memset(head,NULL,sizeof(head));//初始化
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
}
SPFA(n);
for(int i=2;i<=n;i++)
cout<<dis[i]<<endl;
}
return 0;
}
我也把bellman算法写的代码也贴上去把,虽然超时了,但是对于其它数据小一点的题目还是可以过的,在这里分清两种算法
#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAX 9999999
typedef struct
{
int u,v,cost;
}Edge;
Edge edge[200001];
int pre[200001];
int dis[200001];
int m,n;
int original;
bool bellman_ford()
{
for(int i=2;i<=n;i++)
dis[i]=MAX;
for(int i=1;i<=n-1;i++)
for(int j=1;j<=m;j++)
if(dis[edge[j].v]>dis[edge[j].u]+edge[j].cost)
{
dis[edge[j].v]=dis[edge[j].u]+edge[j].cost;
pre[edge[j].v]=edge[j].u;
}
bool flag=true;
for(int i=1;i<=m;i++)
{
if(dis[edge[i].v]>dis[edge[i].u]+edge[i].cost)
{
flag=false;
break;
}
}
return flag;
}
int main()
{
while(cin>>n>>m)
{
pre[1]=0;
for(int i=1; i<=m; i++)
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].cost);
//cin>>edge[i].u>>edge[i].v>>edge[i].cost;
}
if(bellman_ford())
for(int i=2;i<=n;i++)
cout<<dis[i]<<endl;
}
return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。