QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#385312#8136. Rebellious Edgeucup-team3215#WA 87ms3684kbC++202.3kb2024-04-10 17:32:292024-04-10 17:32:31

Judging History

你现在查看的是最新测评结果

  • [2024-04-10 17:32:31]
  • 评测
  • 测评结果:WA
  • 用时:87ms
  • 内存:3684kb
  • [2024-04-10 17:32:29]
  • 提交

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'