算法竞赛入门经典 每日一题(上学要迟到了)

题目描述 

牛牛早上起床一看，自己睡过了，赶紧起床准备去学校，他去学校只有两种方式，坐公交车和步行，牛牛去学校是一条直线，这条直线上总共有n个车站，车站之间的距离都是相等的，每个车站只有一种公交车aiai​，每个公交车只在对应的公交站停车，每个公交车的速度也不一样，第i种公交车过一站的时间需要titi​，并且公交车是单向行驶，只能从左到到右，走路可以任意走，然而牛牛自己步行走一站需要的时间为T，恰好牛牛家和学校都在某一个站点，分别为s和t，问最少需要多少时间牛牛才能到学校？ 

输入描述:

第一行为正整数 n,m,s,t,Tn,m,s,t,T 分别表示直线上的车站个数，公交车种数，牛牛家的位置，学校的位置和牛牛自己步行走一站需要的时间 （n,m≤1×105,1≤s,t≤n,1≤T≤1×104）（n,m≤1×105,1≤s,t≤n,1≤T≤1×104）
第二行有 mm 个正整数表示每种公交车过一站需要的时间为ti（1≤ti≤1×104）ti​（1≤ti​≤1×104）

第三行有 nn 个正整数表示每个车站可停的公交车为ai（1≤ai≤m）ai​（1≤ai​≤m）

输出描述:

输出牛牛从家到学校的最少时间

示例1

输入

复制

5 3 1 5 10
3 2 1
1 3 2 3 1

输出

复制

3

说明

第一种公交车可以从1到5，所以直接乘坐第一种公交车，可以从1到5，第一种公交车没站花费时间为3

看题就可以看出是最短路的题目

那么要如何建图呢：

首先我们要考虑到人可以往出走也可以往回走

所以我们要对人建立双向图

对bus建立的是单向的图

然后对构建好 的图跑dij就行

const int MAX=100010;
struct E{
    int to,w,next;
}e[MAX];
int cnt;
int head[MAX];
int cost[MAX];
int bus[MAX];
void add(int u,int v,int w){
    e[cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}
int a[MAX];
int dist[MAX];
int vis[MAX];
void dij(int s){
    queue<int>q;
    q.push(s);
    dist[s]=0;
    vis[s]=1;
    while (!q.empty()) {
        int u=q.front();
        q.pop();
        for(int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(dist[v]>dist[u]+e[i].w){
                dist[v]=dist[u]+e[i].w;
                if(!vis[v]){
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
        vis[u]=0;
    }
    
    
}
int main(){
    int n,m,s,t,T;
    mms(dist,INF);
    mms(head,-1);
    cin>>n>>m>>s>>t>>T;
    for(int i=1;i<=m;i++){
        cin>>cost[i];
    }
    for(int i=1;i<=n;i++)
        cin>>a[i];
    for(int i=1;i<=n;i++){
        if(i>1) add(i,i-1,T);
        if(i<n) add(i,i+1,T);
        if(bus[a[i]]) add(bus[a[i]],i,cost[a[i]]);
        bus[a[i]]=i;
    }
    dij(s);
    cout<<dist[t];
}