QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#385312 | #8136. Rebellious Edge | ucup-team3215# | WA | 87ms | 3684kb | C++20 | 2.3kb | 2024-04-10 17:32:29 | 2024-04-10 17:32:31 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
using pii = array<int, 2>;
using e = array<int, 3>;
using graph = vector< vector<e> >;
long long prim(const graph & g, int to) {
int n = g.size();
const int INF = 2e9;
vector<int> d(n, INF);
d[0] = 0;
priority_queue<pii, vector<pii>, greater<>> pq;
pq.push({0, 0});
vector<char> visited(n);
vector<int> prev(n, -1);
set<int> path;
for (int _ = 0; _ < 2; ++_) {
while (pq.size()) {
auto [_, u] = pq.top();
pq.pop();
if (visited[u]) {
continue;
}
visited[u] = true;
for (auto [_, v, w] : g[u]) {
if (w < d[v] && !visited[v] && !path.count(v)) {
d[v] = w;
prev[v] = u;
pq.push({d[v], v});
}
}
}
if (path.empty()) {
while (prev[to] != -1) {
path.insert(to);
to = prev[to];
}
path.insert(to);
to = -1;
vector<int> d2(d);
visited.assign(n, false);
d.assign(n, INF);
for (int el : path) {
d[el] = d2[el];
pq.push({d[el], el});
}
}
}
long long ans = 0;
for (int el : d) {
ans += el;
}
return ans;
}
void solve() {
int n, m;
cin >> n >> m;
e spec;
vector<e> es;
for (int i = 0; i < m; ++i) {
int u, v, w;
cin >> u >> v >> w;
--u;--v;
es.push_back({u, v, w});
if (u > v) {
spec = {u, v, w};
}
}
graph g1(n);
for (auto [u, v, w] : es) {
if (v == spec[1]) {
continue;
}
g1[u].push_back({u, v, w});
}
g1[spec[0]].push_back(spec);
graph g2(n);
for (auto [u, v, w] : es) {
if (u == spec[0] && v == spec[1]) {
continue;
}
g2[u].push_back({u, v, w});
}
cout << min(prim(g1, spec[1]), prim(g2, spec[1])) << '\n';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int t;
cin >> t;
while (t--) {
solve();
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3684kb
input:
3 5 6 1 2 4 1 3 2 2 3 0 3 4 1 4 5 1 5 2 1 4 4 1 2 4 1 3 6 1 4 8 4 2 1000000 3 3 1 2 100 2 1 10 2 3 1000
output:
5 18 1100
result:
ok 3 number(s): "5 18 1100"
Test #2:
score: -100
Wrong Answer
time: 87ms
memory: 3632kb
input:
50000 4 5 2 4 998973548 2 3 501271695 4 1 778395982 1 4 32454891 1 2 757288934 4 5 1 4 720856669 2 3 665098951 3 4 407461517 2 1 866914401 1 2 457859826 4 5 1 2 75288664 1 4 624893594 3 2 458973866 2 4 769074655 2 3 834773751 4 5 2 3 237060634 2 4 297307882 3 4 200383586 1 2 42856502 3 1 16574713 4 ...
output:
1291015520 1530420294 1534956009 480300722 1366795927 1541095843 2567005573 858095911 1034153425 793861088 605832428 1051598350 612891589 1265994009 517769091 1899616226 1556463491 93634961 960978736 984886788 1696503797 1002892611 1969660290 1431417780 1515267731 977157479 1937478556 654475526 1401...
result:
wrong answer 7th numbers differ - expected: '2493849488', found: '2567005573'