0
点赞
收藏
分享

微信扫一扫

PAT_甲级_1030 Travel Plan (30分) (C++)【Dijkstra/DFS】


目录

​​1,题目描述​​

​​题目大意​​

​​输入​​

​​输出​​

​​2,思路​​

​​数据结构​​

​​DFS算法设计​​

​​Dijkstra算法设计​​

​​3,代码​​

1,题目描述

PAT_甲级_1030 Travel Plan (30分) (C++)【Dijkstra/DFS】_甲级

Sample Input:

4 5 0 3
0 1 1 20
1 3 2 30
0 3 4 10
0 2 2 20
2 3 1 20

 

Sample Output:

0 2 3 3 40

题目大意

求出图中两点的最短路径。若有多条最短路径,输出花费最小的那条路径;

输入

  1. 第一行:N城市数目(<=500,编号0 - N-1),M路的数目,S起点start,D终点destination;
  2. 接下来M行:表示路的两端连接城市的编号,距离,花费;

输出

  1. 起点到终点的路径, 最短距离, 最少花费

 

2,思路

用Dijkstra算法求出所有最短路径,用DFS遍历所有最短路径,求出最短路径中花费最少的一条(题目保证最短路径中,花费均不相同)。

整体思路类似于这一题​​PAT_甲级_1018 Public Bike Management (30分) (C++)【Dijkstra保存多条最短路径信息+DFS】​​

 

数据结构

将部分数据声明为全局变量,可以减少函数对应的参数,但需更加注意变量是否会无意更改。 

  • int graph[500][500], cost[500][500]:graph存放每条路的用时 cost存放每条路的花费;
  • bool visited[500]:记录是否节点已遍历;
  • int dis[500]:记录起点到其余各点的最短距离;
  • vector<int> path[500]:存放所有节点的所有最短路径,eg.,容器path[2]存放了所有可以到达节点2的最短路径的前驱节点
  • vector<int> temPath, finalPath:temPath临时的最终路径 finalPath最终路径;

 

DFS算法设计

  • 递归的顺序是从终点向原点进行(由于path中存放的是前驱节点);
  • 叶节点(递归出口)为起始点
  • 可以考虑将花费作为参数进行运算,到达叶节点时再更新最少话费;
  • 注意:为了记录最终的最短路径,开始时需将当前的点入栈,当前节点的所有子树遍历完毕后,弹出该节点;

 

Dijkstra算法设计

基本上都是这个思路,很实用,建议熟记。 

  1. visited数组初始化全为false、dis数组初始化全为INT_MAX、graph中两节点未连接部分初始化全为INT_MAX、dis[s](s即算法的起点)初始化为0;
  2. 寻找dis数组中最小的值对应的节点u(当前可达未遍历节点的最短距离);
  3. 遍历所有与u相邻且未访问过的节点v,若dis[v] > dis[u] + graph[u][v],则更新dis[v],并清除此节点v原先记录的所有最短路径(path[v].clear();),并将新的最短前驱节点存入其中(path[v].push_back(u));若dis[v] == dis[u] + graph[u][v],path[v].push_back(u)即可;

 

3,代码

#include<iostream>
#include<vector>
#include<climits>
#include<algorithm>
using namespace std;

bool visited[500];
int graph[500][500], cost[500][500]; //graph存放每条路的用时 cost存放每条路的花费
int dis[500];
vector<int> path[500]; //存放所有节点的所有最短路径
vector<int> temPath, finalPath; //temPath临时的最终路径 finalPath最终路径
int n, m, s, d; //n城市数目 m路数目 s起点 d终点
int minCost = INT_MAX;

void dfs(int end, int co){
temPath.push_back(end);
if(end == s){ //已经从终点到达起点(倒着进行DFS)
if(co < minCost){ //更新最少花费 以及最终路径
finalPath = temPath;
minCost = co;
}
}else{
for(int i = 0; i < path[end].size(); i++){
int e = path[end][i];
dfs(e, co + cost[end][e]);
}
}
temPath.pop_back(); //记得弹出已完全遍历过的节点
}

int main(){
//#ifdef ONLINE_JUDGE
//#else
// freopen("1.txt", "r", stdin);
//#endif

int a, b;
cin>>n>>m>>s>>d;

if(s == d){ //起点即为终点
cout<<s<<' '<<0<<' '<<0;
return 0;
}

fill(graph[0], graph[0] + 500 * 500, INT_MAX);
fill(dis, dis + 500, INT_MAX);
int dis_, cost_;
for(int i = 0; i < m; i++){
scanf("%d%d%d%d", &a, &b, &dis_, &cost_);
graph[a][b] = graph[b][a] = dis_;
cost[a][b] = cost[b][a] = cost_; //注意是无向图cost[a][b] = cost[b][a]
}

dis[s] = 0; //起点先设置dis为0
for(int i = 0; i < n; i++){
int minDis = INT_MAX, u = -1;
for(int j = 0; j < n; j++){
if(visited[j] == false && dis[j] < minDis){
u = j;
minDis = dis[j];
}
}

if(u == -1) break;
visited[u] = true; //赋值号别写成==

for(int v = 0; v < n; v++){
if(visited[v] == false && graph[u][v] < INT_MAX){
if(dis[v] > dis[u] + graph[u][v]){
dis[v] = dis[u] + graph[u][v];
path[v].clear();
path[v].push_back(u);
}else if(dis[v] == dis[u] + graph[u][v]){ //保存多条最短路径
path[v].push_back(u);
}
}
}
}

dfs(d, 0);
for(int i = finalPath.size() - 1; i >= 0; i--){
printf("%d ", finalPath[i]);
}
printf("%d %d", dis[d], minCost);


return 0;
}

 

举报

相关推荐

0 条评论