选择最佳路线(调麻了)

46 阅读4分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 22 天,点击查看活动详情

很简单的一题,但是我寄了,别问为什么,也不知是谁多组测试数据,idx不初始化成0,我不说是谁。

  1. 选择最佳线路

有一天,琪琪想乘坐公交车去拜访她的一位朋友。

由于琪琪非常容易晕车,所以她想尽快到达朋友家。

现在给定你一张城市交通路线图,上面包含城市的公交站台以及公交线路的具体分布。

已知城市中共包含 n� 个车站(编号11~n�)以及 m� 条公交线路。

每条公交线路都是 单向的,从一个车站出发直接到达另一个车站,两个车站之间可能存在多条公交线路。

琪琪的朋友住在 s� 号车站附近。

琪琪可以在任何车站选择换乘其它公共汽车。

请找出琪琪到达她的朋友家(附近的公交车站)需要花费的最少时间。

输入格式

输入包含多组测试数据。

每组测试数据第一行包含三个整数 n,m,s�,�,�,分别表示车站数量,公交线路数量以及朋友家附近车站的编号。

接下来 m� 行,每行包含三个整数 p,q,t�,�,�,表示存在一条线路从车站 p� 到达车站 q�,用时为 t�。

接下来一行,包含一个整数 w�,表示琪琪家附近共有 w� 个车站,她可以在这 w� 个车站中选择一个车站作为始发站。

再一行,包含 w� 个整数,表示琪琪家附近的 w� 个车站的编号。

输出格式

每个测试数据输出一个整数作为结果,表示所需花费的最少时间。

如果无法达到朋友家的车站,则输出 -1。

每个结果占一行。

数据范围

n≤1000,m≤20000�≤1000,�≤20000,
1≤s≤n1≤�≤�,
0<w<n0<�<�,
0<t≤10000<�≤1000

输入样例:

5 8 5
1 2 2
1 5 3
1 3 4
2 4 7
2 5 6
2 3 5
3 5 1
4 5 1
2
2 3
4 3 4
1 2 3
1 3 4
2 3 2
1
1

输出样例:

1
-1

分析

两种方法,第一种方法,反向建边跑一遍dijkstra,第二种方法建立虚拟原点并向num的各个边之间建一条边权为0的边,然后跑一遍dijkstra,我不知道为什么还有那么多人用spfa,是没被卡过嘛???

方法一

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <map>
#include <set>
#include <iomanip>
#include <cmath>
#include <unordered_map>
#include <stack>
#include <queue>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;
typedef pair<int,ll> PII;
typedef pair<string,int> PSI;
typedef stack<int> stk;
ll gcd(ll x,ll y){
    return y?gcd(y,x%y):x;
}
ll qmi(ll x,ll y,int mod){
    ll res=1;
    while(y){if(y&1) res=res*x%mod;y>>=1;x=x*x%mod;}
    return res;
}
const int N=2010,M=20010;
int n,m,s;//n车站m路线,朋友家车站为s
int num;//琪琪家车站数
int id[M];//车站编号
int idx=0,e[M],ne[M],h[N],w[M],dist[N];
inline void add(int a,int b,int c){
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
bool st[N];
inline void dijkstra(){
    memset(dist,0x3f,sizeof dist);
    dist[s]=0;
    priority_queue<PII,vector<PII>,greater<PII>> heap;
    heap.push({0,s});
    while(heap.size()){
        PII t=heap.top();
        heap.pop();
        int distance=t.first,ver=t.second;
        if(st[ver]) continue;
        //cout<<ver<<" ";
        st[ver]=true;
        for(int i=h[ver];~i;i=ne[i]){
            int j=e[i];
           // cout<<j<<" ";
            if(dist[j]>distance+w[i]){
                dist[j]=distance+w[i];
                heap.push({dist[j],j});
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    while(cin>>n>>m>>s){
        idx=0;
        memset(h,-1,sizeof h);
        memset(st,false,sizeof st);
        while(m--){
            int a,b,c;
            cin>>a>>b>>c;
            add(b,a,c);
        }
        cin>>num;
        for(int i=1;i<=num;i++){
            cin>>id[i];
            //add(0,id[i],0);
        }
        dijkstra();
        int res=0x3f3f3f3f;
        for(int i=1;i<=num;i++){
            res=min(res,dist[id[i]]);
        }
        if(res>1e9) cout<<"-1\n";
        else cout<<res<<"\n";
        //cout<<dijkstra()<<"\n";
    }
    return 0;
}

方法二

记好数组要翻倍,md,tle死我了。。。
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <map>
#include <set>
#include <iomanip>
#include <cmath>
#include <unordered_map>
#include <stack>
#include <queue>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;
typedef pair<int,ll> PII;
typedef pair<string,int> PSI;
typedef stack<int> stk;
ll gcd(ll x,ll y){
    return y?gcd(y,x%y):x;
}
ll qmi(ll x,ll y,int mod){
    ll res=1;
    while(y){if(y&1) res=res*x%mod;y>>=1;x=x*x%mod;}
    return res;
}
const int N=2010,M=40010;
int n,m,s;//n车站m路线,朋友家车站为s
int num;//琪琪家车站数
int idx=0,e[M],ne[M],h[M],w[M],dist[M];
inline void add(int a,int b,int c){
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
bool st[N];
inline int dijkstra(int x){
    memset(dist,0x3f,sizeof dist);
    memset(st,false,sizeof st);
    dist[x]=0;
    priority_queue<PII,vector<PII>,greater<PII>> heap;
    heap.push({0,x});
    while(heap.size()){
        PII t=heap.top();
        heap.pop();
        int distance=t.first,ver=t.second;
        if(st[ver]) continue;
        //cout<<ver<<" ";
        st[ver]=true;
        for(int i=h[ver];~i;i=ne[i]){
            int j=e[i];
           // cout<<j<<" ";
            if(dist[j]>distance+w[i]){
                dist[j]=distance+w[i];
                heap.push({dist[j],j});
            }
        }
    }
    if(dist[s]==0x3f3f3f3f) return -1;
    return dist[s];
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    while(cin>>n>>m>>s){
        idx=0;
        memset(h,-1,sizeof h);
        while(m--){
            int a,b,c;
            cin>>a>>b>>c;
            add(a,b,c);
        }
        cin>>num;
        for(int i=1;i<=num;i++){
            int id;
            cin>>id;
            add(0,id,0);
        }
        int t=dijkstra(0);
        cout<<t<<"\n";
    }
    return 0;
}