Pagini recente » Cod sursa (job #1866716) | Cod sursa (job #1228134) | Cod sursa (job #93023) | Cod sursa (job #1233191) | Cod sursa (job #1451712)
#include <cstring>
#include <fstream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
int P, N, M;
vector<pair<int, int> > V[502];
int dist[502][502];
priority_queue<pair<int, int> > Q;
vector<int> W[502];
int D[52];
int minC[52][52][52];
bool is[52];
void Dijkstra(int x)
{
memset(dist[x], 0x3f, sizeof(dist[x]));
dist[x][x] = 0;
Q.push(make_pair(0, x));
while (!Q.empty())
{
auto now = Q.top();
Q.pop();
if (-now.first != dist[x][now.second]) continue;
for (auto edge : V[now.second])
if (-now.first + edge.second < dist[x][edge.first])
{
dist[x][edge.first] = -now.first + edge.second;
Q.push(make_pair(-dist[x][edge.first], edge.first));
}
}
}
int main()
{
ifstream fin("team.in");
ofstream fout("team.out");
fin >> P >> N >> M;
for (int i = 1, nod1, nod2, cost; i <= M; ++i)
{
fin >> nod1 >> nod2 >> cost;
V[nod1].push_back(make_pair(nod2, cost));
V[nod2].push_back(make_pair(nod1, cost));
}
Dijkstra(1);
D[0] = 1;
for (int i = 1; i <= P; ++i)
{
fin >> D[i];
Dijkstra(D[i]);
W[D[i]].push_back(i);
}
memset(minC, 0x3f, sizeof(minC));
for (int s = 1; s <= P; ++s)
for (int i = 1; i <= P - s + 1; ++i)
{
int j = i + s - 1;
for (int k = 0; k <= P; ++k)
{
memset(is, 0, sizeof(is));
for (int l = 1; l <= P; ++l)
if (!is[l])
{
int cand = dist[D[k]][D[l]];
int last = i - 1;
for (auto v : W[D[l]])
{
if (last + 1 <= v - 1)
cand += minC[last + 1][v - 1][l];
is[v] = true;
last = v;
}
if (last + 1 <= j)
cand += minC[last + 1][j][l];
minC[i][j][k] = min(minC[i][j][k], cand);
}
}
}
fout << minC[1][P][0] << '\n';
fin.close();
fout.close();
}