QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#594555 | #2161. The Cost of Speed Limits | thesupermarketisgoingtome | WA | 935ms | 1570180kb | C++20 | 2.2kb | 2024-09-28 07:03:48 | 2024-09-28 07:03:48 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
int n, c;
cin >> n >> c;
vector<vector<pair<int, int>>> g(n);
vector<int> ws;
for (int i = 0; i < n - 1; i++) {
int x, y, w;
cin >> x >> y >> w;
--x; --y;
g[x].emplace_back(y, w);
g[y].emplace_back(x, w);
ws.push_back(w);
}
ranges::sort(ws);
ws.resize(unique(ws.begin(), ws.end()) - ws.begin());
const int W = ws.size();
for (int i = 0; i < n; i++) {
for (auto& [_, w] : g[i]) {
w = ranges::lower_bound(ws, w) - ws.begin();
}
}
vector<int> pe_cost(n, -1);
const int inf = 2e9 + 5;
vector<vector<int>> dp(n, vector<int>(W));
function<void(int, int)> Dfs = [&](int v, int p) {
long long pay_cost = 0;
for (auto [u, w] : g[v]) {
if (u == p) {
continue;
}
pe_cost[u] = w;
Dfs(u, v);
pay_cost += *ranges::min_element(dp[u]) + c;
for (int j = 0; j < W; j++) {
if (dp[v][j] == inf || dp[u][j] == inf) {
dp[v][j] = inf;
} else {
dp[v][j] += dp[u][j];
}
}
}
if (p != -1) {
pay_cost += c;
}
// cerr << "pay cost: " << v << ' ' << pay_cost << endl;
long long pw = (p == -1 ? 0 : pe_cost[v]);
for (int j = 0; j < W; j++) {
if (j < pw) {
dp[v][j] = inf;
} else {
long long pay = ws[j] - ws[pw];
if (dp[v][j] < inf) {
dp[v][j] += pay;
}
dp[v][j] = min((long long) dp[v][j], pay_cost + pay);
}
}
};
Dfs(0, -1);
// for (int i = 0; i < n; i++) {
// for (int j = 0; j < W; j++) {
// cerr << dp[i][j] << " \n"[j == W - 1];
// }
// }
int ans = inf;
for (int j = 0; j < W; j++) {
ans = min(ans, dp[0][j] - (ws[j] - ws[0]));
}
cout << ans << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3528kb
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: 3596kb
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: 3528kb
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: 3780kb
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: 931ms
memory: 1422600kb
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: 927ms
memory: 1570180kb
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: 935ms
memory: 1570140kb
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: 0
Accepted
time: 5ms
memory: 5788kb
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:
1999780001
result:
ok single line: '1999780001'
Test #9:
score: 0
Accepted
time: 19ms
memory: 23368kb
input:
20000 100 1 2 114 2 3 243 3 4 243 4 5 239 5 6 248 6 7 122 7 8 187 8 9 281 9 10 241 10 11 209 11 12 119 12 13 241 13 14 243 14 15 236 15 16 151 16 17 272 17 18 132 18 19 281 19 20 272 20 21 194 21 22 233 22 23 220 23 24 271 24 25 278 25 26 153 26 27 108 27 28 229 28 29 130 29 30 291 30 31 245 31 32 2...
output:
1636347
result:
ok single line: '1636347'
Test #10:
score: -100
Wrong Answer
time: 0ms
memory: 3808kb
input:
1 12345
output:
2000000005
result:
wrong answer 1st lines differ - expected: '0', found: '2000000005'