#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<vector<pair<ll,ll>>>graph;
vector<bool> vis;
void dfs1(vector<ll> &dist, ll x, ll d){
vis[x] = true;
dist[x] = d;
for (auto [y, w] : graph[x]){
if (!vis[y]){
dfs1(dist, y, d + w);
}
}
}
void dfs2(vector<pair<ll,ll>>dist, ll x, ll d){
vis[x] = true;
dist[x].first= d;
for (auto [y, w] : graph[x]){
if (!vis[y]){
dfs2(dist, y, d + w);
}
}
}
ll max_score(int N, int X, int Y, ll K, vector<int> U, vector<int> V, vector<int> W){
graph.assign(N,vector<pair<ll,ll>>());
for(ll i=0;i<(N-1);i++){
graph[U[i]].push_back({V[i],W[i]});
graph[V[i]].push_back({U[i],W[i]});
}
vis.assign(N,false);
vector<ll>distX(N);
dfs1(distX,X,0);
vis.assign(N,false);
vector<ll>distY(N);
dfs1(distY,Y,0);
vis.assign(N,false);
vector<pair<ll,ll>>orden;
for(ll i=0;i<N;i++){
pair<ll,ll>par;
par.first=0;
par.second=i;
orden.push_back(par);
}
dfs2(orden,max_element(distX.begin(),distX.end())-distX.begin(),0);
sort(orden.begin(),orden.end());
ll pos_x;
ll pos_y;
for(ll i=0;i<N;i++){
if(orden[i].second==X){
pos_x=i;
}
else if(orden[i].second==Y){
pos_y=i;
}
}
ll ans=0;
for(ll i=0;i<=pos_x;i++){
for(ll j=pos_x;j<N;j++){
for(ll k=0;k<=pos_y;k++){
for(ll l=pos_y;l<N;l++){
ll suma=0;
for(ll index=0;i<N;index++){
ll v=0;
if(index>=i&&index<=j){
v=max(v,distX[orden[index].second]);
}
else if(index>=k&&index<=l){
v=max(v,distY[orden[index].second]);
}
suma+=v;
}
if(suma<=K){
ans=max(suma,ans);
}
}
}
}
}
return ans;
}
Compilation message
closing.cpp: In function 'll max_score(int, int, int, ll, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:60:5: warning: 'pos_y' may be used uninitialized in this function [-Wmaybe-uninitialized]
60 | for(ll l=pos_y;l<N;l++){
| ^~~
closing.cpp:46:5: warning: 'pos_x' may be used uninitialized in this function [-Wmaybe-uninitialized]
46 | ll pos_x;
| ^~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
851 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1056 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1056 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1056 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1079 ms |
348 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |