#include "closing.h"
#include <bits/stdc++.h>
#define range(it, a, b) for (ll it = a; it < b; it++)
#define all(x) begin(x), end(x)
#define ll long long
#define ull unsigned long long
#define INF64 ((ll) 1 << 60)
#define INF32 ((ll) 1 << 30)
#define uset unordered_set
#define umap unordered_map
#define pqueue priority_queue
using namespace std;
struct TPos {
ll X, Y;
};
bool operator < (const TPos& a, const TPos& b) {
return (max(a.X, a.Y) > max(b.X, b.Y));
}
vector<vector<pair<ll, ll>>> adj;
vector<ll> from_x, from_y;
void dfs_x (ll i, ll cnt) {
from_x[i] = 0;
for (pair<ll, ll> k : adj[i]) {
if (from_x[k.first] == INF64) {
dfs_x(k.first, cnt + k.second);
}
}
}
void dfs_y (ll i, ll cnt) {
from_y[i] = 0;
for (pair<ll, ll> k : adj[i]) {
if (from_y[k.first] == INF64) {
dfs_y(k.first, cnt + k.second);
}
}
}
int max_score(int N, int X, int Y, long long K, vector<int> U, vector<int> V, vector<int> W) {
adj.resize(N);
from_x.resize(N, INF64);
from_y.resize(N, INF64);
range(i, 0, N - 1){
adj[U[i]].push_back({V[i], W[i]});
adj[V[i]].push_back({U[i], W[i]});
}
ll ans = 2*N;
dfs_x(X, 0);
dfs_y(Y, 0);
pqueue<TPos> pq;
ll sum = 0;
range(i, 0, N) {
pq.push({from_x[i], from_y[i]});
sum += max(from_x[i], from_y[i]);
}
while (sum > K) {
TPos t = pq.top(); pq.pop();
if (t.X > t.Y) {
ll dif = t.X - t.Y;
t.X = -INF64;
ans--;
sum -= dif;
}
else {
ll dif = t.Y - t.X;
t.Y = -INF64;
ans--;
sum -= dif;
}
pq.push(t);
}
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
95 ms |
35556 KB |
1st lines differ - on the 1st token, expected: '451', found: '400000' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '6' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '6' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '6' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
1st lines differ - on the 1st token, expected: '6', found: '14' |
2 |
Halted |
0 ms |
0 KB |
- |