题目链接:PAT【甲级】1111
题目简述:给⼀张地图,两个结点中的道路,既有距离属性也有时间属性,要求根据地图推荐两条/一条路线:一条是快速到达路线,一条是短距离的路线。
超时的DFS版本代码【最后一个点超时】
#include<bits/stdc++.h>
using namespace std;
int source, dest, min_time = -1, min_distance = -1, min_intersections = -1, min_dtime = -1;
vector<int> vertexs[500], vis(500, 0), ans_t, ans_d, temp;
vector<vector<int>> times(500, vector<int>(500, -1)), distances(500, vector<int>(500, -1));
void dfs(int beg, int time, int distance, int intersections){
if(beg == dest){
//如果用时相等且站点数最小、用时更短、没有初始化(这里站点数和最小时间总是同时更改所以只需判断一个)
//还有就是要千万注意逻辑短路这个现象,所以这边的条件判断还是要有顺序进行的
if(min_time == -1 || time < min_time || (time == min_time && intersections < min_intersections)){
min_intersections = intersections;
min_time = time;
ans_t.clear();
for (int i = 0; i < temp.size();i++)
ans_t.push_back(temp[i]);
}
if(min_distance == -1 || distance < min_distance || (distance == min_distance && time < min_dtime)){
min_dtime = time;
min_distance = distance;
ans_d.clear();
for (int i = 0; i < temp.size();i++)
ans_d.push_back(temp[i]);
}
return;
}
if((time > min_time && min_time != -1) && (distance > min_distance && min_distance != -1))
return;//剪枝操作
for (int i = 0; i < vertexs[beg].size();i++){
if(vis[vertexs[beg][i]] == 0){
vis[vertexs[beg][i]] = 1;
temp.push_back(vertexs[beg][i]);
dfs(vertexs[beg][i], times[vertexs[beg][i]][beg] + time, distances[vertexs[beg][i]][beg] + distance, intersections + 1);
vis[vertexs[beg][i]] = 0;
temp.pop_back();
}
}
}
int main(){
int n, m, isSame = 1;
int s, e, a, l, t;
cin >> n >> m;
for (int i = 0; i < m;i++){
cin >> s >> e >> a >> l >> t;
vertexs[s].push_back(e);
vertexs[e].push_back(s);
times[s][e] = times[e][s] = t;
distances[s][e] = distances[e][s] = l;
}
cin >> source >> dest;
vis[source] = 1;
dfs(source, 0, 0, 0);
if(ans_d.size() == ans_t.size()){
for (int i = 0; i < ans_t.size();i++){
if(ans_t[i] != ans_d[i]){
isSame = 0;
break;//能省点时间就省点
}
}
}else
isSame = 0;
if(isSame == 0){
printf("Distance = %d: %d", min_distance, source);
for (int i = 0; i < ans_d.size();i++)
printf(" -> %d", ans_d[i]);
printf("\nTime = %d: %d", min_time, source);
for (int i = 0; i < ans_t.size();i++)
printf(" -> %d", ans_t[i]);
}else{
printf("Distance = %d; Time = %d: %d", min_distance, min_time, source);
for (int i = 0; i < ans_t.size();i++)
printf(" -> %d", ans_t[i]);
}
cout << endl;
return 0;
}
哎,可惜超时了。本来以为这个图应该不是很复杂,边数不是很多的那种,没想到最后一个点超时了。
AC的代码
#include<bits/stdc++.h>
using namespace std;
int n, m, source, dest, min_distance, min_time;
vector<int> ans_t, ans_d, dpre(500, -1), tpre(500, -1);
vector<vector<int>> times(500, vector<int>(500, -1)), distances(500, vector<int>(500, -1)), graph(500,vector<int>(500, 0));
void dfs1(int x){
if(x == -1) return;
dfs1(dpre[x]);
ans_d.push_back(x);
}
void dfs2(int x){
if(x == -1) return;
dfs2(tpre[x]);
ans_t.push_back(x);
}
void dijkstra1(int beg){
vector<int> lowcost(500, 0x3f3f3f3f), vis(500, 0), dtime(500, 0x3f3f3f3f);
lowcost[beg] = 0;
dpre[beg] = -1;
for (int i = 0; i < n;i++){
int Min = 0x3f3f3f3f, k = -1;
for (int j = 0; j < n;j++){
if(vis[j] == 0 && lowcost[j] < Min){
Min = lowcost[j];
k = j;
}
}
if(k == -1) break;
vis[k] = 1;
for (int j = 0; j < n;j++){
if(vis[j] == 0 && graph[j][k] == 1){
//如果满足条件则更新信息,这里要注意逻辑短路现象
if(lowcost[j] > lowcost[k] + distances[j][k] || (lowcost[j] == lowcost[k] + distances[j][k] && dtime[k] + times[k][j] < dtime[j])){
dpre[j] = k;
dtime[j] = dtime[k] + times[k][j];
lowcost[j] = lowcost[k] + distances[j][k];
}
}
}
}
min_distance = lowcost[dest];
dfs1(dest);
}
void dijkstra2(int beg){
vector<int> lowcost(500, 0x3f3f3f3f), vis(500, 0), intersections(500, 0x3f3f3f3f);
lowcost[beg] = 0;
tpre[beg] = -1;
for (int i = 0; i < n;i++){
int Min = 0x3f3f3f3f, k = -1;
for (int j = 0; j < n;j++){
if(vis[j] == 0 && lowcost[j] < Min){
Min = lowcost[j];
k = j;
}
}
if(k == -1) break;
vis[k] = 1;
for (int j = 0; j < n;j++){
if(vis[j] == 0 && graph[j][k] == 1){
if(lowcost[j] > lowcost[k] + times[j][k] || (lowcost[j] == lowcost[k] + times[j][k] && intersections[k] + 1 < intersections[j])){
tpre[j] = k;
intersections[j] = intersections[k] + 1;
lowcost[j] = lowcost[k] + times[j][k];
}
}
}
}
min_time = lowcost[dest];
dfs2(dest);
}
int main(){
int s, e, a, l, t, isSame = 1;
cin >> n >> m;
for (int i = 0; i < m;i++){
cin >> s >> e >> a >> l >> t;
graph[s][e] = graph[e][s] = 1;
times[s][e] = times[e][s] = t;
distances[s][e] = distances[e][s] = l;
}
cin >> source >> dest;
dijkstra1(source);
dijkstra2(source);
if(ans_d.size() == ans_t.size()){
for (int i = 0; i < ans_t.size();i++){
if(ans_t[i] != ans_d[i]){
isSame = 0;
break;//能省点时间就省点
}
}
}else
isSame = 0;
if(isSame == 0){
printf("Distance = %d: %d", min_distance, ans_d[0]);
for (int i = 1; i < ans_d.size();i++)
printf(" -> %d", ans_d[i]);
printf("\nTime = %d: %d", min_time, ans_t[0]);
for (int i = 1; i < ans_t.size();i++)
printf(" -> %d", ans_t[i]);
}else{
printf("Distance = %d; Time = %d: %d", min_distance, min_time, ans_t[0]);
for (int i = 1; i < ans_t.size();i++)
printf(" -> %d", ans_t[i]);
}
cout << endl;
return 0;
}
其实这些题目练习熟了还是比较简单的,无非就是这两种;不行就换着来。套着框架写就行了。