Dijkstra Algorithm

题意:
TT 从家里出发,准备乘坐猫猫快线前往喵星机场。猫猫快线分为经济线和商业线两种,它们的速度与价钱都不同。当然啦,商业线要比经济线贵,TT 平常只能坐经济线,但是今天 TT 的魔法猫变出了一张商业线车票,可以坐一站商业线。假设 TT 换乘的时间忽略不计,请你帮 TT 找到一条去喵星机场最快的线路。
输入:
输入包含多组数据。每组数据第一行为 3 个整数 N, S 和 E (2 ≤ N ≤ 500, 1 ≤ S, E ≤ 100),即猫猫快线中的车站总数,起点和终点(即喵星机场所在站)编号。下一行包含一个整数 M (1 ≤ M ≤ 1000),即经济线的路段条数。接下来有 M 行,每行 3 个整数 X, Y, Z (1 ≤ X, Y ≤ N, 1 ≤ Z ≤ 100),表示 TT 可以乘坐经济线在车站 X 和车站 Y 之间往返,其中单程需要 Z 分钟。下一行为商业线的路段条数 K (1 ≤ K ≤ 1000)。接下来 K 行是商业线路段的描述,格式同经济线。所有路段都是双向的,但有可能必须使用商业车票才能到达机场。保证最优解唯一。
输出:
对于每组数据,输出3行。第一行按访问顺序给出 TT 经过的各个车站(包括起点和终点),第二行是 TT 换乘商业线的车站编号(如果没有使用商业线车票,输出"Ticket Not Used",不含引号),第三行是 TT 前往喵星机场花费的总时间。本题不忽略多余的空格和制表符,且每一组答案间要输出一个换行。
输入样例:
4 1 4
4
1 2 2
1 3 3
2 4 4
3 4 5
1
2 4 3
输出样例:
1 2 4
2
5
解题思路:
这道题有两种思路,第一是直接枚举所有的商业线的情况。题目给定了起点与终点,而且要求商业线最多乘坐一次,我们可以直接枚举每一条商业线,计算起点到u的最短路以及v到终点的最短路再加上该商业线所花费的时间。先以起点为源点用Dijkstra算法求一次单源最短路,得到 dis1 数组,然后再以终点为源点再用一次Dijkstra算法求单源最短路,得到 dis2 数组。之后枚举商业线(u, v, w),取 min{dis1[u]+dis2[v]+w, dis1[v]+dis2[u]+w},最终再与不走商业线的答案相比较,取耗费时间最小的情况。还有一种思路是跑一次单源最短路,记录答案 dis[u][0/1],其中dis[u][0] 表示从起点到结点 u 没有经过商业线时的最短路,在松弛的时候可以选择商业线或者经济线。而dis[u][1] 表示从起点到结点 u已经经过商业线后的最短路,在松弛的时候只能选择经济线。
注意事项:
memset函数一般用来对数组或者结构体进行初始化,一般用来赋值为0,1,-1或者极大值(inf):一般用0x3f,因为它是按照字节来进行赋值的,其他值可能会出错。当有多于1条商业线可供使用时(如k条,k>1),我们要求到终点的过程中,使用0∼k条商业线,此时应该使用分层图最短路算法。
总结:
一道利用Dijkstra算法求最短路的问题,注意变形点所在即商业线的加入,因为商业线只用一条,所以可以两次Dijkstra算法以后枚举使用每一条商业线的情况,如果多于1条商业线可供使用时(如k条,k>1),我们要求到终点的过程中,使用0∼k条商业线,此时应该使用分层图最短路算法。
参考代码:

#include <iostream>
#include <utility>
#include <queue>
#include <limits.h>
#include <vector>
#include <cstring>
#include <cstdio>
using namespace std;
const int N=100020;
int n;//num of vertices
int s,t;
int m,k;
struct edge{
    int to, next ,w;
    edge(int a=0,int b=-1,int c=0):to(a),next(b),w(c){}
} e[N];
int head[N],tot=0,vis[N],dis1[N],dis2[N],path1[N],path2[N];
int min_cost=INT_MAX;
int change_station=0;
int change_station2=0;
bool flag=false;
void ini(){
    tot=0;min_cost=INT_MAX;change_station=0;change_station2=0;
 /*   for (int i=1;i<=n;i++) {
        head[i]=-1;
        vis[i]=false;
        dis1[i]=dis2[i]=INT_MAX;
        path1[i]=path2[i]=0;
    }*/
    memset(vis, false, sizeof(vis));
    for (int i = 0; i <= n; i++) {
        head[i] = -1;
        path1[i] = 0;
        path2[i] = 0;
        dis1[i] = 10000000;
        dis2[i] = 10000000;
    }
}
void add(int x,int y,int w){
    e[tot].to = y;
    e[tot].next = head[x];
    e[tot].w = w;
    head[x] = tot;
    tot++;
}



void dijkstra1(int s) {
priority_queue<pair<int,int> > q;
    while(q.size()) q.pop();
    dis1[s]=0;
    path1[s]=-1;
q.push(make_pair(0,s));
    while(q.size()) {
int x = q.top( ).second;
q. pop();
if(vis[x]) continue;
vis[x] = 1;//记录有效的出队次数
for(int i = head[x]; i!=-1; i = e[i] .next) {
 int y = e[i].to, w = e[i].w;
    if(dis1[y] > dis1[x] + w){
dis1[y] = dis1[x] + w;
q.push(make_pair(-dis1[y],y));//入队
        path1[y]=x;
            }
        }
    }
}

void dijkstra2(int s) {
priority_queue<pair<int,int> > q;while(q.size()) q.pop();
dis2[s] = 0;path2[s]=-1;
q.push(make_pair(0,s));
    while(q.size()) {
int x = q.top( ).second;
q. pop();
if(vis[x]) continue;
vis[x] = 1;//记录有效的出队次数
for(int i = head[x]; i!=-1; i = e[i] .next) {
 int y = e[i].to, w = e[i].w;
    if(dis2[y] > dis2[x] + w){
dis2[y] = dis2[x] + w;
q.push(make_pair(-dis2[y],y));//入队
        path2[y]=x;
            }
        }
    }
}


int main(int argc, const char * argv[]) {
    while (scanf("%d%d%d", &n, &s, &t) != EOF) {
        ini();
        vector<int> v;//path
        cin>>m;
        int x,y,z;
        while (m--) {
            cin>>x>>y>>z;
            add(x, y, z);
            add(y,x,z);
        }
        dijkstra1(s);
        memset(vis, false, sizeof(vis));
        dijkstra2(t);
        int k;cin>>k;
        while (k--) {
            cin>>x>>y>>z;
            if(min_cost>dis1[x]+dis2[y]+z){
                change_station=x;
                change_station2=y;
                min_cost=dis1[x]+dis2[y]+z;}
            if(min_cost>dis1[y]+dis2[x]+z){
            change_station=y;
            change_station2=x;
            min_cost=dis1[y]+dis2[x]+z;}
        }
        if(flag) cout<<endl;
        else flag=true;
        if(min_cost>dis1[t]){//不使用商业线
            for (int i=t; i!=s; i=path1[i]) {
                v.push_back(i);
            }
            v.push_back(s);
            for (int i=(int)v.size()-1; i>0; i--)
                printf("%d ", v[i]);
                printf("%d\nTicket Not Used\n%d\n", v[0], dis1[t]);
        }
        else{//使用商业线
            for (int i=change_station; i!=s; i=path1[i]) {
                v.push_back(i);
            }
            v.push_back(s);
            for (int i=change_station2; i!=t; i=path2[i])
            v.insert(v.begin(),i);
            v.insert(v.begin(),t);
            for (int i=(int)v.size()-1; i>0; i--)
                printf("%d ", v[i]);
                printf("%d\n%d\n%d\n", v[0], change_station, min_cost);
        }
        v.clear();
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值