请编写程序求给定正权有向图的单源最短路径长度。图中包含n个顶点,编号为0至n-1,以顶点0作为源点。
输入格式:
输入第一行为两个正整数n和e,分别表示图的顶点数和边数,其中n不超过20000,e不超过1000。接下来e行表示每条边的信息,每行为3个非负整数a、b、c,其中a和b表示该边的端点编号,c表示权值。各边并非按端点编号顺序排列。
输出格式:
输出为一行整数,为按顶点编号顺序排列的源点0到各顶点的最短路径长度(不含源点到源点),每个整数后一个空格。如源点到某顶点无最短路径,则不输出该条路径长度。
输入样例:
4 4
0 1 1
0 3 1
1 3 1
2 0 1
输出样例:
1 1
用的vector的邻接表存储,时间复杂度O(n^2),测试点2直接TLE了,建议换成priority_queue.
#include<bits/stdc++.h>
using namespace std;
#define MaxSize 20001
int dist[MaxSize];
bool visit[MaxSize]={false};
typedef struct enode
{
int v,dis;
}Node;
vector<Node> v[MaxSize];
int vnum,e;
int main()
{
cin>>vnum>>e;
int i;
for(i=0;i<e;i++){
int p,q,length;
cin>>p>>q>>length;
Node temp;
temp.v=q;
temp.dis=length;
v[p].push_back(temp);
}
fill(dist,dist+MaxSize,99999);
dist[0]=0;
int j;
for(i=0;i<vnum;i++){
int u=-1;
int min=100000;
for(j=0;j<vnum;j++){
if(visit[j]==false && dist[j]<min){
u=j;
min=dist[j];
}
}
if(u==-1){
break;
}
visit[u]=true;
for(j=0;j<v[u].size();j++){
int q=v[u][j].v;
if(visit[q]==false&& dist[u]+v[u][j].dis<dist[q]){
dist[q]=dist[u]+v[u][j].dis;
}
}
}
for(i=1;i<vnum;i++){
if(dist[i]!=99999){
cout<<dist[i]<<" ";
}
}
return 0;
}