Fire-Fighting Hero（dijkstra堆优化）

Fire-Fighting Hero

题解

这题目真的恶心，在考理解？还是猜测？
求英雄到各个点的最短路中的最大的
和各个消防队到每个点的最短路中最大的
最后比较

代码

#include<cstdio>
#include<vector>
#include<queue>
#include<utility>
#include<cstring>
using namespace std;
#define MAXN 1020
#define INF 0x3f
typedef long long ll;
//b、dijkstra 堆优化 O（nlogn）

typedef pair<ll, ll> P; //first是到源点最短距离，second是顶点的编号

struct edge { ll to, cost; 
    edge (ll too, ll costt) { to = too; cost = costt; }
};//指向顶点to的权值为cost的边
vector <edge> G[MAXN];//存图 可以理解为二维的，每个G包含若干项，这些项是以当前编号点为起点, 若干项边结构体（edge）的v为终点，cost为权值的边
void addedge (ll u, ll v, ll w) {
    G[u].push_back (edge (v, w)); //添加从u到v权值为w的边
}

ll ma[MAXN][MAXN];
ll kl[MAXN];
ll dis[MAXN][MAXN];
//ll pre[MAXN];//记录前驱节点
ll d[MAXN];//最短距离
ll n, m, s ;//顶点数 边数 起点
//vector<ll> path;//路径
priority_queue<P, vector<P>, greater<P> > que;
void dijkstra () {
    //fill (d, d + n + 1, INF); //fill (pre, pre + n + 1, -1);
    memset(d,INF,sizeof(d));
    d[s] = 0; que.push (P (0, s));
    while (!que.empty()) {//核心
        P p = que.top();  que.pop();
        ll v = p.second;
        if (d[v] < p.first) continue;
        for (ll i = 0; i < G[v].size(); i++) {
            edge e = G[v][i];
            if (d[e.to] > d[v] + e.cost) {
                d[e.to] = d[v] + e.cost;
                que.push (P (d[e.to], e.to));
                //pre[e.to]=v;
            }
        }
    }
}
int main()
{
    int t=1;
    scanf("%d",&t);
    while(t--)
    {
        ll S,k,c;
        memset(ma,INF,sizeof(ma));
        scanf("%lld%lld%lld%lld%lld",&n,&m,&S,&k,&c);
        for(ll i=0;i<k;i++)
        {
            scanf("%lld",&kl[i]);
        }
        ll x,y,z;
        for(ll i=0;i<m;i++)
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            ma[x][y]=ma[y][x]=min(ma[x][y],z);
        }
        for(ll i=1;i<=n;i++)
        {
            G[i].clear();
        }
        for(ll i=1;i<=n;i++)
        {
            for(ll j=1;j<=n;j++)
            {
                if(ma[i][j]<=10010)
                {
                    addedge(i,j,ma[i][j]);
                    //addedge(j,i,ma[i][j]);
                }
            }
        }
        s=S;
        dijkstra();
        ll ans1=-1,ans2=-1;//temp=0x3f3f3f3f;
        for(ll i=1;i<=n;i++)
            ans1=max(ans1,d[i]);
        for(ll i=0;i<k;i++)
        {
            s=kl[i];
            dijkstra();
            for(int j=1;j<=n;j++)
            if(s!=j)dis[s][j]=d[j];
        }
        ll temp=0x3f3f3f3f;
        for(int j=1;j<=n;j++)
        {
            temp=0x3f3f3f3f;
            for(int i=0;i<k;i++)
            {
                temp=min(temp,dis[kl[i]][j]);
            }
            ans2=max(ans2,temp);
        }
        if(m==0)
        {
            printf("0\n");
            continue;
        }
        if(ans1<=c*ans2)printf("%lld\n",ans1);
        else printf("%lld\n",ans2);
    }
    return 0;
}