QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#549721#2161. The Cost of Speed LimitsHKOI0WA 904ms1570044kbC++201.9kb2024-09-06 20:15:092024-09-06 20:15:10

Judging History

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

  • [2024-09-06 20:15:10]
  • 评测
  • 测评结果:WA
  • 用时:904ms
  • 内存:1570044kb
  • [2024-09-06 20:15:09]
  • 提交

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>;

const int inf = 0x3f3f3f3f;

void solve() {
    int n, c;
    cin >> n >> c;
    vector<vector<pii>> e(n);
    vi s(n);
    for (int i = 0; i < n - 1; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        u--;
        v--;
        s[i] = w;
        e[u].emplace_back(v, w);
        e[v].emplace_back(u, w);
    }
    sort(all(s));
    s.erase(unique(all(s)), s.end());
    vector dp(n, vector<int>(sz(s), inf));

    auto dfs = [&](auto&& dfs, int u, int f = -1, int fw = -1) -> void {
        ll sum = 0;
        for (auto [v, w] : e[u])
            if (v != f) {
                dfs(dfs, v, u, w);
                sum += *min_element(all(dp[v]));
            }
        int st = (int)(lower_bound(all(s), fw) - s.begin());
        for (int i = st; i < sz(s); i++) {
            if (fw == -1) dp[u][i] = min((ll)dp[u][i], sum + (ll)c * sz(e[u]));
            else dp[u][i] = min((ll)dp[u][i], sum + s[i] - fw + (ll)c * sz(e[u]));
        }
        for (int i = st; i < sz(s); i++) {
            ll tmp = 0;
            bool ok = true;
            for (auto [v, w] : e[u])
                if (v != f) {
                    if (dp[v][i] == inf) {
                        ok = false;
                        break;
                    }
                    tmp += dp[v][i];
                }
            if (ok) {
                if (fw == -1) dp[u][i] = min((ll)dp[u][i], tmp);
                else dp[u][i] = min((ll)dp[u][i], tmp + s[i] - fw);
            }
        }
    };

    dfs(dfs, 0);
    cout << *min_element(all(dp[0])) << '\n';
}

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

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

13 20
1 8 101
1 9 30
1 2 100
1 3 100
2 4 75
2 5 70
2 6 82
2 7 77
3 10 73
3 11 69
3 12 83
3 13 79

output:

272

result:

ok single line: '272'

Test #2:

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

input:

9 10
1 6 26
2 6 27
3 6 28
4 6 29
5 6 30
7 9 14
8 9 1
9 6 10

output:

60

result:

ok single line: '60'

Test #3:

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

input:

7 64
2 1 194
3 1 187
4 2 158
5 1 42
6 5 101
7 5 80

output:

308

result:

ok single line: '308'

Test #4:

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

input:

6 32
2 1 110
3 2 36
4 1 54
5 3 101
6 5 71

output:

178

result:

ok single line: '178'

Test #5:

score: 0
Accepted
time: 758ms
memory: 1422312kb

input:

20000 100
2273 4097 98627
14155 14055 33506
16060 6363 28081
14840 12695 23379
11520 7892 5831
6633 13834 73944
19218 19341 62064
14392 160 58289
18147 209 46443
16941 5453 55103
11895 12849 31201
10275 1622 71781
19595 6349 14232
19485 10800 9778
10745 13541 44786
18727 15264 25726
5847 12815 43894...

output:

2999800

result:

ok single line: '2999800'

Test #6:

score: 0
Accepted
time: 885ms
memory: 1570044kb

input:

20000 1000
1 2 99998
2 3 99994
3 4 99992
4 5 99989
5 6 99987
6 7 99983
7 8 99982
8 9 99978
9 10 99973
10 11 99970
11 12 99965
12 13 99961
13 14 99957
14 15 99954
15 16 99949
16 17 99946
17 18 99945
18 19 99940
19 20 99935
20 21 99932
21 22 99927
22 23 99923
23 24 99920
24 25 99919
25 26 99918
26 27 ...

output:

2088429

result:

ok single line: '2088429'

Test #7:

score: 0
Accepted
time: 904ms
memory: 1569904kb

input:

20000 1000
1 2 4
2 3 6
3 4 9
4 5 13
5 6 15
6 7 17
7 8 22
8 9 23
9 10 27
10 11 28
11 12 31
12 13 36
13 14 38
14 15 39
15 16 44
16 17 46
17 18 49
18 19 52
19 20 55
20 21 60
21 22 62
22 23 64
23 24 67
24 25 68
25 26 71
26 27 72
27 28 76
28 29 77
29 30 82
30 31 86
31 32 89
32 33 90
33 34 95
34 35 96
35 ...

output:

2098509

result:

ok single line: '2098509'

Test #8:

score: -100
Wrong Answer
time: 5ms
memory: 5460kb

input:

20000 100000
1 4 1
1 5 1
1 6 1
1 7 1
1 8 1
1 9 1
1 10 1
1 11 1
1 12 1
1 13 1
1 14 1
1 15 1
1 16 1
1 17 1
1 18 1
1 19 1
1 20 1
1 21 1
1 22 1
1 23 1
1 24 1
1 25 1
1 26 1
1 27 1
1 28 1
1 29 1
1 30 1
1 31 1
1 32 1
1 33 1
1 34 1
1 35 1
1 36 1
1 37 1
1 38 1
1 39 1
1 40 1
1 41 1
1 42 1
1 43 1
1 44 1
1 45 1...

output:

1061109567

result:

wrong answer 1st lines differ - expected: '1999780001', found: '1061109567'