/* Author : Mychecksdead */
#include<bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long int ll;
typedef long double ld;
#define MOD1 (1000000000+7)
#define MOD (998244353)
#define PI 3.1415926535
#define pb push_back
#define setp() cout << setprecision(15)
#define all(x) x.begin(), x.end()
#define debug(x) cerr << #x << " is " << x << '\n';
const int N = 1e6+100, M = 1e5+10, F = 2147483646, K = 20;
int n, q;
bool case3 = 1, case4 = 1;
pair<int, ll> par[N];
ll w, dp[N];
vector<int> g[N];
vector<pair<int, ll>> edges(N);
multiset<ll> s, dps[N], answers;
ll f(multiset<ll> &a){
if(a.empty()) return 0;
if(a.size() == 1) return *(a.begin());
auto it = a.end(); --it;
ll x = *it; --it; x += *it;
return x;
}
ll dfs(int v, int p){
pair<ll, ll> best = {0, 0};
dp[v] = 0;
ll d = 0;
for(int E: g[v]){
auto e = edges[E];
int u = e.first;
if(u == p) continue;
d = max(d, dfs(u, v));
par[u] = {v, e.second};
dp[v] = max(dp[v], dp[u] + e.second);
if(best.first < dp[u] + e.second){
swap(best.first, best.second);
best.first = dp[u] + e.second;
}else if(best.second < dp[u] + e.second){
best.second = dp[u] + e.second;
}
if(case4)
dps[v].insert(dp[u] + e.second);
}
if(case4)
answers.insert(best.first + best.second);
return max(d, best.first + best.second);
}
void solve(){
cin >> n >> q >> w;
for(int i = 0; i < n - 1; ++i){
int u, v; ll x; cin >> u >> v >> x;
g[u].pb(2*i);
g[v].pb(2*i+1);
if(u != 1 && v != 1) case3 = 0;
if(v/2 != u && u/2 != v) case4 = 0;
s.insert(x);
edges[i*2] = {v, x};
edges[i*2+1] = {u, x};
}
if(case4) dfs(1, 0);
ll last = 0;
for(;q--;){
ll d, x; cin >> d >> x;
d = (d + last) % (n-1);
x = (x + last) % w;
ll old_x = edges[2*d].second;
edges[2*d].second = edges[2*d+1].second = x;
if(case3){
s.erase(s.find(old_x));
s.insert(x);
auto it = s.end(); --it;
if(s.size()==1) last = *it;
else{
last = *it;
it--;
last += *it;
}
}else if(case4){
int v = edges[2*d].first;
if(par[v].first != edges[2*d+1].first) v = edges[2*d].first;
ll cur = 0, delta = x - old_x;
if(dps[v].size() > 0) cur = *(--dps[v].end());
par[v].second = x;
while(v != 1){
cur += par[v].second;
v = par[v].first;
answers.erase(answers.find(f(dps[v])));
dps[v].erase(dps[v].find(cur - delta));
dps[v].insert(cur);
answers.insert(f(dps[v]));
}
assert(v!=1);
last = *(--answers.end());
}else{
// answers.clear();
last = dfs(1, 0);
// last = *(--answers.end());
}
cout << last << '\n';
}
}
int main(){
cin.tie(0); ios::sync_with_stdio(0);
int T = 1, aa;
// cin >> T;aa=T;
while(T--){
// cout << "Case #" << aa-T << ": ";
solve();
cout << '\n';
}
return 0;
}
Compilation message
diameter.cpp: In function 'int main()':
diameter.cpp:137:16: warning: unused variable 'aa' [-Wunused-variable]
137 | int T = 1, aa;
| ^~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
43 ms |
86356 KB |
Output is correct |
2 |
Correct |
42 ms |
86444 KB |
Output is correct |
3 |
Correct |
43 ms |
86464 KB |
Output is correct |
4 |
Correct |
50 ms |
86460 KB |
Output is correct |
5 |
Correct |
43 ms |
86388 KB |
Output is correct |
6 |
Correct |
39 ms |
86356 KB |
Output is correct |
7 |
Correct |
45 ms |
86440 KB |
Output is correct |
8 |
Correct |
43 ms |
86424 KB |
Output is correct |
9 |
Correct |
44 ms |
86444 KB |
Output is correct |
10 |
Correct |
41 ms |
86348 KB |
Output is correct |
11 |
Correct |
53 ms |
86340 KB |
Output is correct |
12 |
Correct |
45 ms |
86424 KB |
Output is correct |
13 |
Correct |
41 ms |
86348 KB |
Output is correct |
14 |
Correct |
40 ms |
86404 KB |
Output is correct |
15 |
Correct |
41 ms |
86356 KB |
Output is correct |
16 |
Correct |
45 ms |
86340 KB |
Output is correct |
17 |
Correct |
41 ms |
86440 KB |
Output is correct |
18 |
Correct |
42 ms |
86348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
43 ms |
86356 KB |
Output is correct |
2 |
Correct |
42 ms |
86444 KB |
Output is correct |
3 |
Correct |
43 ms |
86464 KB |
Output is correct |
4 |
Correct |
50 ms |
86460 KB |
Output is correct |
5 |
Correct |
43 ms |
86388 KB |
Output is correct |
6 |
Correct |
39 ms |
86356 KB |
Output is correct |
7 |
Correct |
45 ms |
86440 KB |
Output is correct |
8 |
Correct |
43 ms |
86424 KB |
Output is correct |
9 |
Correct |
44 ms |
86444 KB |
Output is correct |
10 |
Correct |
41 ms |
86348 KB |
Output is correct |
11 |
Correct |
53 ms |
86340 KB |
Output is correct |
12 |
Correct |
45 ms |
86424 KB |
Output is correct |
13 |
Correct |
41 ms |
86348 KB |
Output is correct |
14 |
Correct |
40 ms |
86404 KB |
Output is correct |
15 |
Correct |
41 ms |
86356 KB |
Output is correct |
16 |
Correct |
45 ms |
86340 KB |
Output is correct |
17 |
Correct |
41 ms |
86440 KB |
Output is correct |
18 |
Correct |
42 ms |
86348 KB |
Output is correct |
19 |
Correct |
168 ms |
86624 KB |
Output is correct |
20 |
Correct |
161 ms |
86552 KB |
Output is correct |
21 |
Correct |
176 ms |
86664 KB |
Output is correct |
22 |
Correct |
197 ms |
86704 KB |
Output is correct |
23 |
Correct |
1104 ms |
87108 KB |
Output is correct |
24 |
Correct |
1151 ms |
87132 KB |
Output is correct |
25 |
Correct |
1237 ms |
87232 KB |
Output is correct |
26 |
Correct |
1412 ms |
87736 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
47 ms |
86380 KB |
Output is correct |
2 |
Correct |
47 ms |
86432 KB |
Output is correct |
3 |
Correct |
46 ms |
86396 KB |
Output is correct |
4 |
Correct |
49 ms |
86668 KB |
Output is correct |
5 |
Correct |
99 ms |
87248 KB |
Output is correct |
6 |
Correct |
45 ms |
86360 KB |
Output is correct |
7 |
Correct |
49 ms |
86476 KB |
Output is correct |
8 |
Correct |
44 ms |
86420 KB |
Output is correct |
9 |
Correct |
43 ms |
86512 KB |
Output is correct |
10 |
Correct |
49 ms |
86732 KB |
Output is correct |
11 |
Correct |
105 ms |
87288 KB |
Output is correct |
12 |
Correct |
42 ms |
86844 KB |
Output is correct |
13 |
Correct |
44 ms |
86836 KB |
Output is correct |
14 |
Correct |
46 ms |
86848 KB |
Output is correct |
15 |
Correct |
66 ms |
87180 KB |
Output is correct |
16 |
Correct |
113 ms |
87636 KB |
Output is correct |
17 |
Correct |
99 ms |
95008 KB |
Output is correct |
18 |
Correct |
111 ms |
94984 KB |
Output is correct |
19 |
Correct |
103 ms |
94948 KB |
Output is correct |
20 |
Correct |
127 ms |
95176 KB |
Output is correct |
21 |
Correct |
241 ms |
95620 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
114 ms |
175480 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
5074 ms |
97276 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
43 ms |
86356 KB |
Output is correct |
2 |
Correct |
42 ms |
86444 KB |
Output is correct |
3 |
Correct |
43 ms |
86464 KB |
Output is correct |
4 |
Correct |
50 ms |
86460 KB |
Output is correct |
5 |
Correct |
43 ms |
86388 KB |
Output is correct |
6 |
Correct |
39 ms |
86356 KB |
Output is correct |
7 |
Correct |
45 ms |
86440 KB |
Output is correct |
8 |
Correct |
43 ms |
86424 KB |
Output is correct |
9 |
Correct |
44 ms |
86444 KB |
Output is correct |
10 |
Correct |
41 ms |
86348 KB |
Output is correct |
11 |
Correct |
53 ms |
86340 KB |
Output is correct |
12 |
Correct |
45 ms |
86424 KB |
Output is correct |
13 |
Correct |
41 ms |
86348 KB |
Output is correct |
14 |
Correct |
40 ms |
86404 KB |
Output is correct |
15 |
Correct |
41 ms |
86356 KB |
Output is correct |
16 |
Correct |
45 ms |
86340 KB |
Output is correct |
17 |
Correct |
41 ms |
86440 KB |
Output is correct |
18 |
Correct |
42 ms |
86348 KB |
Output is correct |
19 |
Correct |
168 ms |
86624 KB |
Output is correct |
20 |
Correct |
161 ms |
86552 KB |
Output is correct |
21 |
Correct |
176 ms |
86664 KB |
Output is correct |
22 |
Correct |
197 ms |
86704 KB |
Output is correct |
23 |
Correct |
1104 ms |
87108 KB |
Output is correct |
24 |
Correct |
1151 ms |
87132 KB |
Output is correct |
25 |
Correct |
1237 ms |
87232 KB |
Output is correct |
26 |
Correct |
1412 ms |
87736 KB |
Output is correct |
27 |
Correct |
47 ms |
86380 KB |
Output is correct |
28 |
Correct |
47 ms |
86432 KB |
Output is correct |
29 |
Correct |
46 ms |
86396 KB |
Output is correct |
30 |
Correct |
49 ms |
86668 KB |
Output is correct |
31 |
Correct |
99 ms |
87248 KB |
Output is correct |
32 |
Correct |
45 ms |
86360 KB |
Output is correct |
33 |
Correct |
49 ms |
86476 KB |
Output is correct |
34 |
Correct |
44 ms |
86420 KB |
Output is correct |
35 |
Correct |
43 ms |
86512 KB |
Output is correct |
36 |
Correct |
49 ms |
86732 KB |
Output is correct |
37 |
Correct |
105 ms |
87288 KB |
Output is correct |
38 |
Correct |
42 ms |
86844 KB |
Output is correct |
39 |
Correct |
44 ms |
86836 KB |
Output is correct |
40 |
Correct |
46 ms |
86848 KB |
Output is correct |
41 |
Correct |
66 ms |
87180 KB |
Output is correct |
42 |
Correct |
113 ms |
87636 KB |
Output is correct |
43 |
Correct |
99 ms |
95008 KB |
Output is correct |
44 |
Correct |
111 ms |
94984 KB |
Output is correct |
45 |
Correct |
103 ms |
94948 KB |
Output is correct |
46 |
Correct |
127 ms |
95176 KB |
Output is correct |
47 |
Correct |
241 ms |
95620 KB |
Output is correct |
48 |
Runtime error |
114 ms |
175480 KB |
Execution killed with signal 6 |
49 |
Halted |
0 ms |
0 KB |
- |