QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#543787#5102. Dungeon CrawlerHKOI0TL 1ms3976kbC++172.9kb2024-09-01 20:16:292024-09-01 20:16:30

Judging History

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

  • [2024-09-01 20:16:30]
  • 评测
  • 测评结果:TL
  • 用时:1ms
  • 内存:3976kb
  • [2024-09-01 20:16:29]
  • 提交

answer

#include <bits/stdc++.h>
#define sz(v) (int)v.size()
#define all(v) v.begin(), v.end()

using namespace std;
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;

template<class T>
ostream& operator<<(ostream& os, const vector<T>& v) {
    for (auto x : v) os << x << ' ';
    os << '\n';
    return os;
}

void solve() {
    int n,q; cin >> n >> q;

    ll distAll = 0;
    vector<vector<pair<int, ll>>> g(n);
    for (int i = 0; i < n - 1; i++) {
        int u,v,w; cin >> u >> v >> w; u--; v--;
        g[u].emplace_back(v, w);
        g[v].emplace_back(u, w);
        distAll += w;
    }

    distAll *= 2;

    vector<vector<int>> moveTo(n, vector<int>(n, -1));
    vector<vector<int>> moveFrom(n, vector<int>(n, -1));
    for (int root = 0; root < n; root++) {
        function<void(int, int)> dfs = [&](int u, int p) {
            moveTo[u][root] = p;
            moveFrom[root][u] = p;
            for (auto& [v, w] : g[u]) {
                if (v == p) continue;
                dfs(v, u);
            }
        };
        dfs(root, root);
    }

    vector<vector<ll>> distOf(n, vector<ll>(n, 0));
    for (int root = 0; root < n; root++) {
        distOf[root][root] = 0;
        function<void(int, int)> dfs = [&](int u, int p) {
            for (auto& [v, w] : g[u]) {
                if (v == p) continue;
                distOf[root][v] = distOf[root][u] + w;
                dfs(v, u);
            }
        };
        dfs(root, root);
    }

    auto getLCA = [&](int root, int u, int v) {
        int cur = root;
        while (moveFrom[u][cur] == moveFrom[v][cur])
            cur = moveFrom[u][cur];
        return cur;
    };

    while (q--) {
        int start, key, trap; cin >> start >> key >> trap; start--; key--; trap--;

        const int KEY_TRAP_LCA = getLCA(start, key, trap);
        if (KEY_TRAP_LCA == trap) {
            cout << "impossible\n";
            continue;
        }
        
        ll ans = LLONG_MAX;
        auto dfs = [&](auto&& dfs, int u, int p, int key_x_lca, ll dist) -> void {
            ll triplePath = 0;
            if (moveTo[KEY_TRAP_LCA][u] == moveTo[KEY_TRAP_LCA][key])
                triplePath = distOf[KEY_TRAP_LCA][key_x_lca];

            ans = min(ans, -dist + 2 * triplePath);

            for (auto& [v, w] : g[u]) {
                if (v == p) continue;
                const bool isKeyPath = (moveFrom[key][u] == v);
                if (isKeyPath)
                    dfs(dfs, v, u, v, dist + w);
                else
                    dfs(dfs, v, u, key_x_lca, dist + w);
            }
        };
        dfs(dfs, start, start, start, 0);

        cout << distAll + ans << '\n';
    }
}

signed main() {
#ifndef LOCAL
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
#endif
    cout << fixed << setprecision(10);
    int T = 1;
    // cin >> T;
    while (T--) solve();
    return 0;
}

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 3548kb

input:

22 5
1 2 7
2 3 5
3 4 8
3 6 6
3 7 3
2 5 6
5 8 2
8 9 11
2 10 16
1 11 12
11 12 4
11 13 9
1 14 25
14 15 4
15 16 5
15 17 6
15 18 1
15 19 8
14 20 7
20 21 9
20 22 17
1 19 9
1 9 19
1 8 9
1 9 8
2 22 11

output:

316
293
293
impossible
314

result:

ok 5 lines

Test #2:

score: 0
Accepted
time: 1ms
memory: 3976kb

input:

100 100
1 2 289384
2 3 930887
2 4 692778
4 5 636916
4 6 747794
4 7 238336
4 8 885387
8 9 760493
8 10 516650
8 11 641422
8 12 202363
8 13 490028
8 14 368691
8 15 520060
8 16 897764
16 17 513927
16 18 180541
16 19 383427
16 20 89173
16 21 455737
16 22 5212
16 23 595369
16 24 702568
16 25 956430
16 26 ...

output:

103526917
103484292
106288816
104379596
104405611
104775512
105434682
105291604
103838430
105371370
104677980
104175650
105894571
104509242
103971939
105376499
105223283
104153426
105082245
105413188
104130613
104800548
106846555
104138329
103769253
105456754
104044745
104385328
106973740
105460460
...

result:

ok 100 lines

Test #3:

score: 0
Accepted
time: 0ms
memory: 3832kb

input:

10 6
1 2 4
2 3 7
2 4 8
4 5 6
4 6 4
4 7 6
4 8 7
8 9 3
8 10 10
3 8 1
10 4 7
1 7 3
7 2 9
8 10 3
4 1 6

output:

99
78
97
87
88
93

result:

ok 6 lines

Test #4:

score: 0
Accepted
time: 0ms
memory: 3788kb

input:

10 9
9 2 5
9 1 6
9 4 97
9 7 2
9 8 42
9 10 11
9 6 77
9 3 14
9 5 9
4 7 10
7 3 8
8 7 9
1 4 8
10 7 4
7 1 2
10 1 5
10 7 2
8 4 10

output:

352
427
impossible
443
418
427
418
418
407

result:

ok 9 lines

Test #5:

score: 0
Accepted
time: 0ms
memory: 3532kb

input:

9 9
2 3 48
9 5 31
7 3 97
4 3 16
1 7 24
5 3 82
8 2 51
6 4 33
1 2 8
3 6 8
1 6 3
9 5 6
2 6 4
5 6 1
9 6 4
2 8 9
4 9 2

output:

530
643
impossible
530
impossible
561
impossible
595
627

result:

ok 9 lines

Test #6:

score: 0
Accepted
time: 0ms
memory: 3556kb

input:

8 9
1 7 51
7 6 86
2 3 62
8 4 72
5 6 17
4 1 75
3 1 41
2 3 7
5 8 4
6 1 3
8 6 2
4 2 7
8 5 6
2 1 5
7 1 6
6 7 8

output:

551
impossible
524
558
579
impossible
551
705
524

result:

ok 9 lines

Test #7:

score: 0
Accepted
time: 0ms
memory: 3592kb

input:

9 9
5 4 13
9 2 10
1 9 25
7 6 34
4 2 77
3 8 67
8 1 57
6 9 100
6 4 1
4 1 7
3 2 9
4 9 7
7 9 3
6 2 1
2 8 4
8 6 2
6 5 9

output:

517
545
impossible
530
483
517
642
584
impossible

result:

ok 9 lines

Test #8:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

10 10
2 4 26
9 8 39
4 5 88
6 3 70
7 6 7
10 4 41
8 3 57
1 6 15
5 6 9
2 8 6
3 9 1
5 7 8
4 7 8
7 6 4
2 7 3
6 8 2
5 4 10
4 8 9
1 5 9

output:

impossible
496
529
441
531
415
566
529
441
523

result:

ok 10 lines

Test #9:

score: 0
Accepted
time: 0ms
memory: 3540kb

input:

10 9
3 2 6
2 1 18
8 1 44
1 9 42
6 3 72
10 8 46
7 10 93
5 3 11
4 10 13
7 3 1
8 2 5
7 5 4
5 10 8
3 1 4
6 4 3
10 1 6
5 10 8
2 10 4

output:

impossible
550
584
impossible
483
impossible
504
impossible
489

result:

ok 9 lines

Test #10:

score: -100
Time Limit Exceeded

input:

2000 199998
126 244 481188299
718 1159 740107180
1327 1250 248943862
977 1092 780169400
826 27 932696654
1668 638 478193038
229 174 176675890
1251 646 843918836
102 1973 593920932
236 218 165399894
760 151 890198591
232 502 10739184
1961 1409 45917915
548 1840 974742709
1096 630 280975617
1110 1048 ...

output:

1266421864327
impossible
1003453161105
1017793822920
1056758437569
impossible
1249128162612
1233756636475
1354563020262
1275484665657
impossible
impossible
1644448395794
impossible
impossible
impossible
1305598243001
1730425595360
1090858373772
1180211385304
1235543994987
1894692656465
impossible
12...

result: