PAT甲级Dijkstra总结

目前PAT甲级中涉及Dijkstra的题目有如下几题：

1003 Emergency (25)

1018 Public Bike Management (30)

1030 Travel Plan (30)

1087 All Roads Lead to Rome (30)

1111 Online Map (30)

1.题目只涉及求出两点之间的最短距离，不需要再根据其余特征筛选最短距离，也就是最原始的Dijkstra：

（1）利用邻接矩阵存储：

//此函数中n为顶点数
void Dijkstra(int s){//s为起点
	fill(d,d+n,inf); //fill将整个d数组赋值为inf
	memset(vis,0,sizeof(vis));
	d[s]=0;//将起点设置为可达
	for(int i=0;i<n;i++){//循环n次
		int u=-1,minn=inf;
		//找到未访问过的顶点中距离最小的
		for(int j=0;j<n;j++){
			if(!vis[j]&&d[j]<minn){
				u=j;
				minn=d[j];
			}
		}
		if(u==-1) return;//如果找不到表明剩下的点和起点s不连通
		vis[u]=1;//标记已访问
		for(int v=0;v<n;v++){
			//如果v未访问并且可达，而且以u为中介点可以使d[v]更小就更新值，如果你并没有初始化
			//g矩阵为inf的话，只需要把下面 inf改成 0就可以了
			if(!vis[v]&&g[u][v]!=inf){
				if(d[u]+g[u][v]<d[v]){
					d[v] = d[u]+g[u][v];
				}
			}
		}
	}
}

 （2）利用邻接表存储

//此函数中n为顶点数
void Dijkstra(int s){//s为起点
	fill(d,d+n,inf); //fill将整个d数组赋值为inf
	memset(vis,0,sizeof(vis));
	d[s]=0;//将起点设置为可达
	for(int i=0;i<n;i++){//循环n次
		int u=-1,minn=inf;
		//找到未访问过的顶点中距离最小的
		for(int j=0;j<n;j++){
			if(!vis[j]&&d[j]<minn){
				u=j;
				minn=d[j];
			}
		}
		if(u==-1) return;//如果找不到表明剩下的点和起点s不连通
		vis[u]=1;//标记已访问
		for(int j=0;j<Adj[u].size();j++){
			//如果v未访问，而且以u为中介点可以使d[v]更小就更新值
			int v=Adj[u][j].v;
			if(!vis[v]&&d[u]+Adj[u][j].dis<d[v]){
				d[v] = d[u]+Adj[u][j].dis;
			}
		}
	}
}

２．题目需要输出最短路径：

我们需要增加一个path数组来存储路径即可：

//此函数中n为顶点数
void Dijkstra(int s){//s为起点
	//初始化路径
	for(int i=0;i<maxn;i++) path[i]=i;
	fill(d,d+n,inf); //fill将整个d数组赋值为inf
	memset(vis,0,sizeof(vis));
	d[s]=0;//将起点设置为可达
	for(int i=0;i<n;i++){//循环n次
		int u=-1,minn=inf;
		//找到未访问过的顶点中距离最小的
		for(int j=0;j<n;j++){
			if(!vis[j]&&d[j]<minn){
				u=j;
				minn=d[j];
			}
		}
		if(u==-1) return;//如果找不到表明剩下的点和起点s不连通
		vis[u]=1;//标记已访问
		for(int v=0;v<n;v++){
			//如果v未访问并且可达，而且以u为中介点可以使d[v]更小就更新值
			if(!vis[v]&&g[u][v]!=inf){
				if(d[u]+g[u][v]<d[v]){
					d[v] = d[u]+g[u][v];
					path[v]=u;//该结点保存的值是到达他的结点，用于递归遍历
				}
			}
		}
	}
}
//递归用于输出路径,e是终点，s是起点
void dfs(int e){
	if(e==s){
		cout<<e<<" ";
		return;
	}
	dfs(path[e]);
	cout<<e<<" ";
}

3.题目要求如果最短路径条数不唯一，根据第二指标确定路径，以及如何用DFS进行第二指标的筛选：

void Dijkstra(int s){
	fill(d,d+maxn,inf);
	fill(vis,vis+maxn,false);
	dis[s]=0;
	for(int i=0;i<n;i++){
		int u=-1,minn=inf;
		for(int j=1;j<=n;j++){
			if(vis[j]==false&&d[j]<minn){
				u=j;
				minn=d[j];
			}
		}
		if(u==-1) return;
		vis[u]=true;
		for(int v=1;v<=n;v++){
			if(vis[v]==false&&g[u][v]!=0){
				if(d[u]+g[u][v]<dis[v]){
					d[v]=d[u]+g[u][v];
					path[v].clear();//找到更小的路径清空原先较大的路径
					path[v].push_back(u);//把更小的前继结点加入
				}else if(d[u]+g[u][v]==d[v]){
					path[v].push_back(u);
				}
			}
		}
	}
}
//v是终点
void dfs(int v){
	if(v==beginn){
		temp.push_back(v);
		/*
			中间代码段根据第二指标选取路径
		*/
		temp.pop_back();
		return;
	}
	temp.push_back(v);
	for(int i=0;i<path[v].size();i++){
		dfs(pre[v][i]);
	}
	temp.pop_back();
}

4.题目要求输出最短路径条数：

void Dijkstra(int s){
	fill(d,d+maxn,inf);
	fill(vis,vis+maxn,false);
	dis[s]=0;
	num[s]=1;//num数组保存起点到其余点路径最短路的条数，初始化起点到其余可达点都有一条路
	for(int i=0;i<n;i++){
		int u=-1,minn=inf;
		for(int j=1;j<=n;j++){
			if(vis[j]==false&&d[j]<minn){
				u=j;
				minn=d[j];
			}
		}
		if(u==-1) return;
		vis[u]=true;
		for(int v=1;v<=n;v++){
			if(vis[v]==false&&g[u][v]!=0){
				if(d[u]+g[u][v]<dis[v]){
					d[v]=d[u]+g[u][v];
					num[v]=num[u];//存在更小的路径，则等于前继结点的条数
				}else if(d[u]+g[u][v]==d[v]){
					num[v]+=num[u];//存在最短路相同，把前继结点的条数加过来
				}
			}
		}
	}
}