이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define x first
#define y second
using ll = long long;
using namespace std;
struct triple {
int x, y;
ll z;
};
int main() {
int n, m;
cin >> n >> m;
vector<vector<pair<int, ll>>> g(n);
vector<vector<triple>> g1(n);
for (int i = 0; i < m; i++) {
int u, v, c;
ll p;
cin >> u >> v >> c >> p; u--; v--; c--;
g1[u].push_back({v, c, p});
g1[v].push_back({u, c, p});
}
vector<int> cnt(m, 0);
vector<ll> sum(m, 0);
for (int i = 0; i < n; i++) {
for (auto [u, c, p] : g1[i]) {
cnt[c]++;
sum[c] += p;
}
for (auto [u, c, p] : g1[i]) {
if (cnt[c] > 1) {
g[i].emplace_back(u, min(p, sum[c] - p));
} else {
g[i].emplace_back(u, 0);
}
}
for (auto [u, c, p] : g1[i]) {
cnt[c]--;
sum[c] -= p;
}
}
set<pair<ll, int>> st;
vector<ll> dist(n, INT64_MAX / 2);
vector<int> used(n, 0);
dist[0] = 0;
for (int i = 0; i < n; i++) {
st.emplace(dist[i], i);
}
while (!st.empty()) {
auto [d, s] = *st.begin();
st.erase(st.begin());
used[s] = 1;
for (auto [i, w] : g[s]) {
if (!used[i]) {
st.erase({dist[i], i});
dist[i] = min(dist[i], d + w);
st.emplace(dist[i], i);
}
}
}
if (dist[n - 1] == INT64_MAX / 2) {
cout << "-1\n";
} else {
cout << dist[n - 1] << "\n";
}
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |