#include "closing.h"
#include<bits/stdc++.h>
using namespace std;
#define int long long int
int32_t max_score(int32_t n, int32_t X, int32_t Y, int k, vector<int32_t> U, vector<int32_t> V, vector<int32_t> W) {
int ans = 0;
vector<vector<pair<int, int>>> adj(n+1);
vector<int> vis(n, 0), distx(n, -1), disty(n, -1);
for (int i = 0; i < n-1; i++) {
adj[U[i]].push_back({V[i], W[i]});
adj[V[i]].push_back({U[i], W[i]});
}
set<tuple<int, int, int>> pq; // let 0 be x, 1 be y
distx[X] = 0;
disty[Y] = 0;
pq.insert({0, X, 0});
pq.insert({0, Y, 1});
while(!pq.empty()) {
auto [d, a, ind] = *pq.begin();
pq.erase(*pq.begin());
if (vis[a] == 0 && distx[a] != -1 && disty[a] != -1) {
if (ind == 0) {
pq.erase({disty[a], a, 1});
pq.insert({disty[a]-distx[a], a, 1});
} else {
pq.erase({distx[a], a, 0});
pq.insert({distx[a]-disty[a], a, 0});
}
}
vis[a] = d;
if (k < d) {
break;
} else {
k -= d;
ans++;
}
if (ind == 0) {
for (auto [b, w] : adj[a]) {
if (distx[b] != -1) {
continue;
}
distx[b] = distx[a]+w;
if (vis[b] >= distx[b]) {
pq.insert({0, b, 0});
} else {
pq.insert({(distx[b]-vis[b]), b, 0});
}
}
} else {
for (auto [b, w] : adj[a]) {
if (disty[b] != -1) {
continue;
}
disty[b] = disty[a]+w;
if (vis[b] >= disty[b]) {
pq.insert({0, b, 1});
} else {
pq.insert({(disty[b]-vis[b]), b, 1});
}
}
}
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
75 ms |
24804 KB |
Output is correct |
2 |
Correct |
78 ms |
24916 KB |
Output is correct |
3 |
Correct |
49 ms |
2904 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '7' |
3 |
Halted |
0 ms |
0 KB |
- |