L2-001. 紧急救援 dijkstra,适合模版

https://www.patest.cn/contests/gplt/L2-001

dijkstra,在求最短路的基础上增加了权值,权值越大越好。

#include <bits/stdc++.h>
using namespace std;

const int maxn=505;
const int INF=0x3f3f3f3f;

int n,m,st,ed;
int G[maxn][maxn]; //城市与城市之间的距离 
int Val[maxn];   //每个城市的救援队数量
int dis[maxn];   //距离起点的最短距离 
int w[maxn];    //距离起点能获得的最多救援队
int num[maxn];  //从起点出发有几条路能达到
bool vis[maxn];
int pre[maxn];

void Dijkstra(){

  dis[st]=0;
  w[st]=Val[st];
  num[st]=1;
  for (int i=0;i<n;i++){
    int u=-1;
    int min_dis=INF;
    for (int j=0;j<n;j++){
      if (!vis[j]&&dis[j]<min_dis){
        u=j;
        min_dis=dis[j];
      }
    }
    if (u==-1) break;
    vis[u]=1;
    for (int v=0;v<n;v++){
      if (!vis[v]&&G[u][v]!=INF){
        if (dis[v]>dis[u]+G[u][v]){
          dis[v]=dis[u]+G[u][v];
          num[v]=num[u];
          w[v]=Val[v]+w[u];
          pre[v]=u;
        }
        else
        if (dis[v]==dis[u]+G[u][v]){
          num[v]+=num[u];
          if (w[v]<w[u]+Val[v]){
            w[v]=w[u]+Val[v];
            pre[v]=u;
          }
        }
      }
    }
  }
}


void printPath(int v){
  if (v==st){
    cout << v;
    return;
  }
  printPath(pre[v]);
  cout << " " << v;
}

int main(){

  cin >> n >> m >> st >> ed;

  fill(G[0],G[0]+maxn*maxn,INF);
  fill(dis,dis+maxn,INF);
  for (int i=0;i<n;i++){
    cin >> Val[i];
  }
  for (int i=0;i<m;i++){
    int u,v,c;
    cin >> u >> v >> c;
    G[u][v]=c;
    G[v][u]=c;
  }

  Dijkstra();

  cout << num[ed] << " " << w[ed] << endl;

  printPath(ed);
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值