2021年度训练联盟热身训练赛第三场 I.Slow Leak(最短路)

题目链接
题意：

给出一个无向图，起点为 $1$, 终点为 $n$, 每次最多只能走$d$ 的距离，图上有一些点，到达这些点可以补充能量，使得下次能继续走 $d$ 的距离。问起点到终点的最短路。

题解：

因为 $n$ 很小，所以可以先用 $floyd$ 算出两两之间的最短路。假如起点到终点的最短路 $\le$ $d$ ,那么直接输出答案。

否则，可以推出，途中必定要经过那些点去补充能量，那么这个过程可以看做是 起点 - 加油站 -加油站 - … -终点。

也就是说，我们要以最短的距离去往加油站，那么我们重新建个图，把起点和终点和加油站两两建边，边权为之前用 $floyd$ 跑出来的最短路。最后再跑一次 $dij$ 即可。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=5e2+5;
const int mod=1e9+7;
const ll inf=0x3f3f3f3f3f3f3f3f;
const double eps=1e-6;
ll d[MAXN][MAXN];
int n;
int p[MAXN];
void solve()
{
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
            }
        }
    }
}
int head[MAXN];
int vis[MAXN];
ll dis[MAXN];
int cnt;
struct edge
{
    int to;
    ll cost;
    int next;
}e[450005];
struct node
{
    int pos;
    ll val;
    friend bool operator<(node a,node b)
    {
        return a.val>b.val;
    }
};
void add(int u,int v,ll w)
{
    e[cnt].to=v;
    e[cnt].cost=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}
void dij(int s)
{
    memset(vis,0,sizeof vis);
    memset(dis,inf,sizeof dis);
    priority_queue<node> q;
    q.push(node{s,0});
    dis[s]=0;
    while(!q.empty())
    {
        node now=q.top();
        q.pop();
        if(vis[now.pos]) continue;
        vis[now.pos]=1;
        for(int i=head[now.pos];i!=-1;i=e[i].next)
        {
            int v=e[i].to;
            if(dis[v]>now.val+e[i].cost)
            {
                dis[v]=now.val+e[i].cost;
                q.push(node{v,dis[v]});
            }
        }
    }
}
int main()
{
    memset(head,-1,sizeof head);
    int m,t,q;
    cin>>n>>m>>t>>q;
    for(int i=1;i<=t;i++)
    {
        cin>>p[i];
    }
    memset(d,inf,sizeof d);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        d[u][v]=d[v][u]=w;
    }
    solve();
    if(d[1][n]<=q)
    {
        printf("%lld\n",d[1][n]);
        return 0;
    }
    p[t+1]=1;
    p[t+2]=n;
    t+=2;
  
    for(int i=1;i<=t;i++)
    {
        for(int j=i+1;j<=t;j++)
        {
            if(d[p[i]][p[j]]<=q)
            {
                add(p[i],p[j],d[p[i]][p[j]]);
                add(p[j],p[i],d[p[i]][p[j]]);
                
            }
        }
    }
    dij(1);
    if(dis[n]<inf) printf("%lld\n",dis[n]);
    else printf("stuck\n");
}