7-6 单源最短路径 (10 分)

7-6 单源最短路径 (10 分)

请编写程序求给定正权有向图的单源最短路径长度。图中包含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

本来我的写法是可以的，但奈何这一题的内存限制为20MB，如果图矩阵用了二维数组的话内存指定就超限了

#include<iostream>
#include<math.h>
using namespace std;

#define MAX 0xffffff

int edge[20009][20009];//图矩阵 
int dist[20009];//原点到i点的最短距离,这个数组也是需要输出的 
int fan[20009];//是否已经被访问过 

int n,e;

void Dijkstra(int v){
	fill(dist,dist + 20009,MAX);
	dist[v] = 0;
	
	for(int i = 0;i < n;i ++){
		int mark = -1,minx = MAX;
		for(int j = 0;j < n;j ++){
			if(minx > dist[j] && fan[j] == 0){//找到dist数组中距离最小并且未被访问的点 
				minx = dist[j];
				mark = j;
			}
		}
		if(mark == -1) break;
		fan[mark] = 1;
		for(int k = 0;k < n;k ++){
			if(fan[k] == 0 && edge[mark][k] != MAX){//未被访问且连通 
				if(dist[k] > edge[mark][k] + dist[mark]){
					dist[k] = edge[mark][k] + dist[mark];
				}
			}
		}
	}
	
}

int main(){
	cin >> n >> e;
	for(int i = 0;i < n;i ++){
		for(int j = 0;j < n;j ++){
			edge[i][j] = MAX;
		}
	}
	for(int i = 0;i < e;i ++){
		int x,y,a;
		cin >> x >> y >> a;
		edge[x][y] = a;
	}
	Dijkstra(0);
	for(int i = 1;i < n;i ++){
		if(dist[i] != MAX){
			cout << dist[i] << " ";
		}
	}
}

下面这个是把图矩阵改为了二维动态数组写的，就过了

#include<iostream>
#include <algorithm>
#include<string.h>
#include<vector>
using namespace std;

#define INF 0xffffff

struct node{
	int x;
	int dis;
}; 

vector<node> a[20005];
int n,e,v,u;
int dist[20005]; //从起点到i点的最短路径
int visit[20005]; //记录是否访问 

void Dijkstra(int v){
	fill(dist,dist + 20005,INF);
	dist[v] = 0;
	for(int i = 0;i < n;i ++){
		u = -1; int minx = INF;
		for(int j = 0;j < n;j ++){
			if(visit[j] == 0 && dist[j] < minx){
				u = j;
				minx = dist[j];
			}
		}
		
		if(u == -1) break;
		visit[u] = 1;
		
		for(int i = 0;i < a[u].size();i ++){
		    int x = a[u][i].x;
			if(visit[x] == 0 && dist[u] + a[u][i].dis < dist[x]){
				dist[x] = dist[u] + a[u][i].dis;
			}
		}
	}
	
}

int main(){
	cin >> n >> e;
	int b1,b2,b3;
	memset(visit,0,sizeof(visit));
	for(int i = 0;i < e;i ++){
		cin >> b1 >> b2 >> b3;
		struct node temp = {b2,b3};
		a[b1].push_back(temp);
	} 
	Dijkstra(v);
	for(int i = 1;i < n;i ++){
		if(dist[i] != INF){
			cout << dist[i] << " ";
		}
	} 
}