#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+5;
vector<tuple<int, ll, ll> > g[MAXN];
pair<ll, int> mprof[MAXN][2];
ll dp[MAXN];
ll respf[MAXN], somaf;
int n, q;
void melhora(int cur, pair<ll, int> val) {
if(val.first>=mprof[cur][0].first) {
mprof[cur][1]=mprof[cur][0];
mprof[cur][0]=val;
}
else if(val.first>mprof[cur][1].first) {
mprof[cur][1]=val;
}
}
void predfs(int cur, int p) {
for(auto aresta : g[cur]) {
int viz; ll peso, rpeso;
tie(viz, peso, rpeso)=aresta;
if(viz==p) continue;
predfs(viz, cur);
dp[cur]+=dp[viz]+rpeso;
melhora(cur, {mprof[viz][0].first+peso, mprof[viz][0].second});
}
}
void dfs(int cur, int p) {
respf[1]=max(respf[1], dp[cur]);
respf[2]=max(respf[2], dp[cur]+mprof[cur][0].first);
// printf("dp[%d]= %lld // mprof[%d][0]= %lld, %d\n", cur, dp[cur], cur, mprof[cur][0].first, mprof[cur][0].second);
for(auto aresta : g[cur]) {
int viz; ll peso, rpeso;
tie(viz, peso, rpeso)=aresta;
if(viz==p) continue;
ll dpc=dp[cur], dpv=dp[viz];
auto mprofc0=mprof[cur][0], mprofc1=mprof[cur][1];
auto mprofv0=mprof[viz][0], mprofv1=mprof[viz][1];
dp[cur]-=(dp[viz]+rpeso);
dp[viz]+=(dp[cur]+peso);
if(mprof[cur][0].second==mprof[viz][0].second) mprof[cur][0]=mprof[cur][1];
melhora(viz, {mprof[cur][0].first+rpeso, mprof[cur][0].second});
dfs(viz, cur);
dp[cur]=dpc; dp[viz]=dpv;
mprof[cur][0]=mprofc0; mprof[cur][1]=mprofc1;
mprof[viz][0]=mprofv0; mprof[viz][1]=mprofv1;
}
}
int main() {
scanf("%d", &n);
for(int i=1; i<n; i++) {
int a, b, c, d; scanf("%d %d %d %d", &a, &b, &c, &d);
g[a].push_back({b, c, d}); g[b].push_back({a, d, c});
somaf+=c; somaf+=d;
}
predfs(1, 1);
dfs(1, 1);
scanf("%d", &q);
while(q--) {
int val; scanf("%d", &val);
printf("%lld\n", somaf-respf[val]);
}
}
Compilation message
designated_cities.cpp: In function 'int main()':
designated_cities.cpp:61:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &n);
~~~~~^~~~~~~~~~
designated_cities.cpp:63:24: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
int a, b, c, d; scanf("%d %d %d %d", &a, &b, &c, &d);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
designated_cities.cpp:71:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &q);
~~~~~^~~~~~~~~~
designated_cities.cpp:73:17: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
int val; scanf("%d", &val);
~~~~~^~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Incorrect |
7 ms |
5112 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Correct |
314 ms |
26448 KB |
Output is correct |
3 |
Correct |
380 ms |
49272 KB |
Output is correct |
4 |
Correct |
285 ms |
26488 KB |
Output is correct |
5 |
Correct |
289 ms |
26152 KB |
Output is correct |
6 |
Correct |
417 ms |
29816 KB |
Output is correct |
7 |
Correct |
260 ms |
25276 KB |
Output is correct |
8 |
Correct |
450 ms |
50148 KB |
Output is correct |
9 |
Correct |
238 ms |
24216 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Incorrect |
312 ms |
26188 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Incorrect |
7 ms |
5112 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Correct |
314 ms |
26448 KB |
Output is correct |
3 |
Correct |
380 ms |
49272 KB |
Output is correct |
4 |
Correct |
285 ms |
26488 KB |
Output is correct |
5 |
Correct |
289 ms |
26152 KB |
Output is correct |
6 |
Correct |
417 ms |
29816 KB |
Output is correct |
7 |
Correct |
260 ms |
25276 KB |
Output is correct |
8 |
Correct |
450 ms |
50148 KB |
Output is correct |
9 |
Correct |
238 ms |
24216 KB |
Output is correct |
10 |
Correct |
6 ms |
5112 KB |
Output is correct |
11 |
Incorrect |
312 ms |
26188 KB |
Output isn't correct |
12 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Incorrect |
7 ms |
5112 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |