이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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 = i + 1; j < n; j++) {
if (i == j) {
continue;
}
dist[i][j].resize(deg[i]);
for (int k = 0; k < deg[i]; k++) {
dist[i][j][k] = vector<long long>(deg[j], inf);
}
}
}
vector<vector<long long>> d(n);
for (int i = 0; i < n; i++) {
d[i].resize(deg[i]);
}
vector<int> ia(n), ib(n);
set<tuple<long long, int, int>> st;
for (int ver = 0; ver < n; ver++) {
for (int e = 0; e < deg[ver]; e++) {
for (int i = 0; i < n; i++) {
for (int j = 0; j < deg[i]; j++) {
d[i][j] = inf;
}
ia[i] = -1;
ib[i] = -1;
}
int to = g[ver][e].first;
int w = g[ver][e].second;
d[to][idx[to][ver]] = w;
st.clear();
st.emplace(d[to][idx[to][ver]], to, idx[to][ver]);
ia[to] = idx[to][ver];
auto Update = [&](int x, int y, long long z) {
if (ia[x] == y || ib[x] == y) {
st.erase(st.find({d[x][y], x, y}));
}
d[x][y] = z;
if (ia[x] == y) {
return;
}
if (ib[x] == y) {
if (d[x][y] < d[x][ia[x]]) {
swap(ia[x], ib[x]);
}
return;
}
if (ia[x] == -1) {
ia[x] = y;
st.emplace(d[x][y], x, y);
return;
}
if (ib[x] == -1) {
if (d[x][y] < d[x][ia[x]]) {
swap(ia[x], ib[x]);
ia[x] = y;
} else {
ib[x] = y;
}
st.emplace(d[x][y], x, y);
return;
}
if (d[x][y] >= d[x][ia[x]] && d[x][y] < d[x][ib[x]]) {
st.erase(st.find({d[x][ib[x]], x, ib[x]}));
ib[x] = y;
st.emplace(d[x][y], x, y);
return;
}
if (d[x][y] < d[x][ia[x]]) {
st.erase(st.find({d[x][ib[x]], x, ib[x]}));
swap(ia[x], ib[x]);
ia[x] = y;
st.emplace(d[x][y], x, y);
}
};
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) {
Update(to, idx[to][v], d[v][lst] + w);
assert((int) st.size() <= 2 * n);
}
}
}
for (int i = ver + 1; i < n; i++) {
if (i == ver) {
continue;
}
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, vector<vector<int>>(2)));
vector<vector<int>> cc(n, vector<int>(2));
for (int from = 0; from < n; from++) {
for (int to = 0; to < n; to++) {
if (from == to) {
continue;
}
vector<pair<int, int>> order;
for (int i = 0; i < deg[from]; i++) {
for (int j = 0; j < deg[to]; j++) {
order.emplace_back(i, j);
}
}
if (from <= to) {
sort(order.begin(), order.end(), [&](pair<int, int> a, pair<int, int> b) {
return dist[from][to][a.first][a.second] < dist[from][to][b.first][b.second];
});
} else {
sort(order.begin(), order.end(), [&](pair<int, int> a, pair<int, int> b) {
return dist[to][from][a.second][a.first] < dist[to][from][b.second][b.first];
});
}
int total = 0;
for (auto& p : order) {
if (cc[p.first][0] >= 2 || cc[p.second][1] >= 2) {
continue;
}
good[from][to][0].push_back(p.first);
good[from][to][1].push_back(p.second);
cc[p.first][0] += 1;
cc[p.second][1] += 1;
total += 1;
if (total >= 5) {
break;
}
}
for (int i = 0; i < deg[from]; i++) {
cc[i][0] = 0;
}
for (int i = 0; i < deg[to]; i++) {
cc[i][1] = 0;
}
}
}
vector<vector<int>> f(l);
for (int i = 0; i < l; i++) {
if (i > 0) {
for (int j : good[seq[i - 1]][seq[i]][1]) {
f[i].push_back(j);
}
}
if (i + 1 < l) {
for (int j : good[seq[i]][seq[i + 1]][0]) {
f[i].push_back(j);
}
}
sort(f[i].begin(), f[i].end());
f[i].erase(unique(f[i].begin(), f[i].end()), f[i].end());
}
vector<long long> dp(n, inf);
for (int i = 0; i < f[0].size(); i++) {
for (int j = 0; j < f[1].size(); j++) {
if (seq[0] <= seq[1]) {
dp[j] = min(dp[j], dist[seq[0]][seq[1]][f[0][i]][f[1][j]]);
} else {
dp[j] = min(dp[j], dist[seq[1]][seq[0]][f[1][j]][f[0][i]]);
}
}
}
vector<long long> new_dp(n);
vector<long long> val(n);
for (int p = 2; p < l; p++) {
for (int i = 0; i < f[p].size(); i++) {
new_dp[i] = inf;
}
for (int i = 0; i < f[p - 1].size(); i++) {
val[i] = inf;
}
{
long long mn = inf;
for (int i = 0; i < f[p - 1].size(); i++) {
val[i] = min(val[i], mn);
mn = min(mn, dp[i]);
}
}
{
long long mn = inf;
for (int i = f[p - 1].size() - 1; i >= 0; i--) {
val[i] = min(val[i], mn);
mn = min(mn, dp[i]);
}
}
for (int i = 0; i < f[p - 1].size(); i++) {
if (val[i] != inf) {
for (int j = 0; j < f[p].size(); j++) {
if (seq[p - 1] <= seq[p]) {
new_dp[j] = min(new_dp[j], val[i] + dist[seq[p - 1]][seq[p]][f[p - 1][i]][f[p][j]]);
} else {
new_dp[j] = min(new_dp[j], val[i] + dist[seq[p]][seq[p - 1]][f[p][j]][f[p - 1][i]]);
}
}
}
}
for (int i = 0; i < deg[seq[p]]; i++) {
dp[i] = new_dp[i];
}
}
long long res = inf;
for (int i = 0; i < deg[seq.back()]; i++) {
res = min(res, dp[i]);
}
cout << (res == inf ? -1 : res) << '\n';
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
wild_boar.cpp: In function 'int main()':
wild_boar.cpp:202:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
202 | for (int i = 0; i < f[0].size(); i++) {
| ~~^~~~~~~~~~~~~
wild_boar.cpp:203:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
203 | for (int j = 0; j < f[1].size(); j++) {
| ~~^~~~~~~~~~~~~
wild_boar.cpp:214:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
214 | for (int i = 0; i < f[p].size(); i++) {
| ~~^~~~~~~~~~~~~
wild_boar.cpp:217:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
217 | for (int i = 0; i < f[p - 1].size(); i++) {
| ~~^~~~~~~~~~~~~~~~~
wild_boar.cpp:222:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
222 | for (int i = 0; i < f[p - 1].size(); i++) {
| ~~^~~~~~~~~~~~~~~~~
wild_boar.cpp:234:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
234 | for (int i = 0; i < f[p - 1].size(); i++) {
| ~~^~~~~~~~~~~~~~~~~
wild_boar.cpp:236:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
236 | for (int j = 0; j < f[p].size(); j++) {
| ~~^~~~~~~~~~~~~
# | 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... |