L2-001. 紧急救援
最短路变形。。。。
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
int const MAX = 505;
int const INF = 0x3fffffff;
int mp[MAX][MAX], val[MAX], path[MAX], dis[MAX], re[MAX], totval[MAX], pathnum[MAX];
//mp 边集 //val 救援队数量
//path 记录定点u -> v中的u //dis s->i的路径长度
//re //totval总救援队
//pathnum s->i的路径数;
bool vis[MAX]; //记录是否被访问;
int n, m, s, d;
void Dijkstra(int v0)
{
for(int i = 0; i < n; i++)
dis[i] = INF;
vis[v0] = true;
dis[v0] = 0;
totval[v0] = val[v0];
pathnum[v0] = 1;
for(int i = 0; i < n; i++)
{
if(mp[v0][i] != INF && i != v0)
{
dis[i] = mp[v0][i];
path[i] = v0;
totval[i] = val[v0] + val[i];
pathnum[i] = 1;
}
}
for(int i = 0; i < n - 1; i++)
{
int mi = INF, mival = 0, u = v0;
for(int j = 0; j < n; j++)
{
if(!vis[j] && dis[j] < mi)
{
mi = dis[j];
u = j;
}
}
vis[u] = true;
for(int j = 0; j < n; j++)
{
if(!vis[j])
{
if(dis[u] + mp[u][j] < dis[j])
{
pathnum[j] = pathnum[u];
dis[j] = dis[u] + mp[u][j];
totval[j] = totval[u] + val[j];
path[j] = u;
}
else if(dis[u] + mp[u][j] == dis[j])
{
pathnum[j] += pathnum[u];
if(totval[j] < totval[u] + val[j])
{
totval[j] = totval[u] + val[j];
path[j] = u;
}
}
}
}
}
}
int main()
{
scanf("%d %d %d %d", &n, &m, &s, &d);
for(int i = 0; i < n; i++)
scanf("%d", &val[i]);
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
mp[i][j] = INF;
int x, y, l;
for(int i = 0; i < m; i++)
{
scanf("%d %d %d", &x, &y, &l);
mp[x][y] = min(mp[x][y], l);
mp[y][x] = mp[x][y];
}
Dijkstra(s);
int num = 0, cur = d;
while(cur != s)
{
re[num ++] = cur;
cur = path[cur];
}
re[num ++] = s;
printf("%d %d\n", pathnum[d], totval[d]);
for(int i = num - 1; i > 0; i--)
printf("%d ", re[i]);
printf("%d\n", re[0]);
}