This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int n, m, t, l;
cin >> n >> m >> t >> l;
vector<vector<pair<int, int>>> g(n);
for (int i = 0; i < m; i++) {
int x, y, z;
cin >> x >> y >> z;
--x; --y;
g[x].emplace_back(y, z);
g[y].emplace_back(x, z);
}
vector<int> seq(l);
for (int i = 0; i < l; i++) {
cin >> seq[i];
--seq[i];
}
{
int p, x;
cin >> p >> x;
--p; --x;
seq[p] = x;
}
vector<int> deg(n);
for (int i = 0; i < n; i++) {
deg[i] = (int) g[i].size();
}
vector<vector<int>> idx(n, vector<int>(n));
for (int i = 0; i < n; i++) {
for (int j = 0; j < deg[i]; j++) {
idx[i][g[i][j].first] = j;
}
}
const long long inf = (long long) 1e18;
vector<vector<vector<vector<long long>>>> dist(n, vector<vector<vector<long long>>>(n));
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
dist[i][j].resize(deg[i]);
for (int k = 0; k < deg[i]; k++) {
dist[i][j][k] = vector<long long>(deg[j], inf);
}
}
}
for (int ver = 0; ver < n; ver++) {
for (int e = 0; e < deg[ver]; e++) {
vector<vector<long long>> d(n);
for (int i = 0; i < n; i++) {
d[i] = vector<long long>(deg[i], inf);
}
int to = g[ver][e].first;
int w = g[ver][e].second;
d[to][idx[to][ver]] = w;
set<tuple<long long, int, int>> st;
st.emplace(d[to][idx[to][ver]], to, idx[to][ver]);
while (!st.empty()) {
auto it = st.begin();
int v = get<1>(*it);
int lst = get<2>(*it);
st.erase(it);
for (int i = 0; i < deg[v]; i++) {
if (i == lst) {
continue;
}
int to = g[v][i].first;
int w = g[v][i].second;
if (d[to][idx[to][v]] > d[v][lst] + w) {
if (d[to][idx[to][v]] != inf) {
st.erase(st.find({d[to][idx[to][v]], to, idx[to][v]}));
}
d[to][idx[to][v]] = d[v][lst] + w;
st.emplace(d[to][idx[to][v]], to, idx[to][v]);
}
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < deg[i]; j++) {
dist[ver][i][e][j] = d[i][j];
}
}
}
}
vector<vector<vector<vector<int>>>> good(n, vector<vector<vector<int>>>(n));
for (int from = 0; from < n; from++) {
for (int to = 0; to < n; to++) {
if (from == to) {
continue;
}
good[from][to].resize(deg[from]);
vector<int> order(deg[to]);
iota(order.begin(), order.end(), 0);
for (int i = 0; i < deg[from]; i++) {
sort(order.begin(), order.end(), [&](int a, int b) {
return dist[from][to][i][a] < dist[from][to][i][b];
});
for (int j = 0; j < min(deg[to], 5); j++) {
good[from][to][i].push_back(order[j]);
}
}
}
}
vector<long long> dp(deg[seq[1]], inf);
for (int i = 0; i < deg[seq[0]]; i++) {
for (int j = 0; j < deg[seq[1]]; j++) {
dp[j] = min(dp[j], dist[seq[0]][seq[1]][i][j]);
}
}
for (int p = 2; p < l; p++) {
vector<long long> new_dp(deg[seq[p]], inf);
vector<long long> val(deg[seq[p - 1]], inf);
for (int i = 0; i < (int) val.size(); i++) {
for (int j = 0; j < (int) val.size(); j++) {
if (i != j) {
val[i] = min(val[i], dp[j]);
}
}
}
for (int i = 0; i < deg[seq[p - 1]]; i++) {
for (int j : good[seq[p - 1]][seq[p]][i]) {
new_dp[j] = min(new_dp[j], val[i] + dist[seq[p - 1]][seq[p]][i][j]);
}
}
swap(dp, new_dp);
}
long long res = *min_element(dp.begin(), dp.end());
cout << (res == inf ? -1 : res) << '\n';
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |