#include "bits/stdc++.h"
using namespace std;
#define FAST ios_base::sync_with_stdio(false); cin.tie(0);
#define pb push_back
#define eb emplace_back
#define ins insert
#define f first
#define s second
#define cbr cerr<<"hi\n"
#define mmst(x, v) memset((x), v, sizeof ((x)))
#define siz(x) ll(x.size())
#define all(x) (x).begin(), (x).end()
#define lbd(x,y) (lower_bound(all(x),y)-x.begin())
#define ubd(x,y) (upper_bound(all(x),y)-x.begin())
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
inline long long rand(long long x, long long y) { return rng() % (y+1-x) + x; } //inclusive
string inline to_string(char c) {string s(1,c);return s;} template<typename T> inline T gcd(T a,T b){ return a==0?llabs(b):gcd(b%a,a); }
using ll=long long;
using ld=long double;
#define FOR(i,s,e) for(ll i=s;i<=ll(e);++i)
#define DEC(i,s,e) for(ll i=s;i>=ll(e);--i)
using pi=pair<ll,ll>; using spi=pair<ll,pi>; using dpi=pair<pi,pi>;
long long LLINF = 1e18;
int INF = 1e9+1e6;
#define MAXN (200006)
ll n, mx, P[MAXN], A[MAXN], C[MAXN], ans;
bitset<MAXN> vis, in;
vector<int> stk, cy;
vector<int> v[MAXN];
void cycle(int x) {
vis[x]=1;
stk.eb(x);
if(!vis[P[x]]) cycle(P[x]);
else {
while(1) {
if(stk.empty()) {
cy.clear();
break;
}
ll y=stk.back(); stk.pop_back();
cy.eb(y);
if(y==P[x]) break;
}
}
}
map<ll,ll> *mp[MAXN];
void dfs(int x) {
mp[x] = new map<ll,ll>;
for(auto i:v[x]) {
dfs(i);
if(mp[x]->size() < mp[i]->size()) mp[x] = mp[i];
}
for(auto i:v[x]) if(mp[x] != mp[i]) {
for(auto j:*mp[i]) (*mp[x])[j.f] += j.s;
}
(*mp[x])[mx] += C[x];
(*mp[x])[A[x]] -= C[x];
(*mp[x])[A[x]-1] += C[x];
auto it = mp[x]->find(A[x]);
while(it->s < 0) {
auto tmp = next(it);
assert(tmp != mp[x]->end());
tmp->s += it->s;
mp[x]->erase(it);
it = tmp;
}
if(it->s == 0) mp[x]->erase(it);
}
int main() {
FAST
cin>>n;
FOR(i,1,n) cin>>P[i]>>A[i]>>C[i];
mx = *max_element(A+1, A+n+1) + 1;
FOR(i,1,n) A[i] = mx - A[i];
FOR(i,1,n) if(P[i]^i) in[P[i]]=1;
FOR(i,1,n) if(!in[i]) {
stk.clear(), cy.clear();
cycle(i);
if(cy.empty()) continue;
sort(all(cy), [](int x,int y){return A[x]>A[y];});
FOR(i,0,siz(cy)-2) {
P[cy[i]] = cy[i+1];
}
P[cy.back()]=cy.back();
}
FOR(i,1,n) if(P[i] ^ i) v[P[i]].eb(i);
FOR(i,1,n) if(P[i] == i) {
dfs(i);
ans += (*mp[i])[mx];
}
cout<<ans<<'\n';
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4940 KB |
Output is correct |
2 |
Correct |
3 ms |
4940 KB |
Output is correct |
3 |
Correct |
4 ms |
4940 KB |
Output is correct |
4 |
Correct |
3 ms |
4940 KB |
Output is correct |
5 |
Correct |
11 ms |
6580 KB |
Output is correct |
6 |
Correct |
8 ms |
6092 KB |
Output is correct |
7 |
Correct |
7 ms |
5964 KB |
Output is correct |
8 |
Correct |
9 ms |
6604 KB |
Output is correct |
9 |
Correct |
8 ms |
6092 KB |
Output is correct |
10 |
Correct |
7 ms |
5964 KB |
Output is correct |
11 |
Correct |
6 ms |
5708 KB |
Output is correct |
12 |
Correct |
8 ms |
6416 KB |
Output is correct |
13 |
Correct |
6 ms |
6308 KB |
Output is correct |
14 |
Correct |
8 ms |
5964 KB |
Output is correct |
15 |
Correct |
7 ms |
5964 KB |
Output is correct |
16 |
Correct |
10 ms |
7004 KB |
Output is correct |
17 |
Correct |
11 ms |
6348 KB |
Output is correct |
18 |
Correct |
7 ms |
5708 KB |
Output is correct |
19 |
Correct |
8 ms |
6220 KB |
Output is correct |
20 |
Correct |
7 ms |
5996 KB |
Output is correct |
21 |
Correct |
6 ms |
6092 KB |
Output is correct |
22 |
Correct |
7 ms |
6092 KB |
Output is correct |
23 |
Correct |
6 ms |
5836 KB |
Output is correct |
24 |
Correct |
8 ms |
6296 KB |
Output is correct |
25 |
Correct |
7 ms |
6092 KB |
Output is correct |
26 |
Correct |
7 ms |
6348 KB |
Output is correct |
27 |
Correct |
8 ms |
6220 KB |
Output is correct |
28 |
Correct |
8 ms |
6348 KB |
Output is correct |
29 |
Correct |
7 ms |
6476 KB |
Output is correct |
30 |
Correct |
8 ms |
6476 KB |
Output is correct |
31 |
Correct |
8 ms |
6476 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4940 KB |
Output is correct |
2 |
Correct |
3 ms |
4940 KB |
Output is correct |
3 |
Correct |
4 ms |
4940 KB |
Output is correct |
4 |
Correct |
3 ms |
4940 KB |
Output is correct |
5 |
Correct |
11 ms |
6580 KB |
Output is correct |
6 |
Correct |
8 ms |
6092 KB |
Output is correct |
7 |
Correct |
7 ms |
5964 KB |
Output is correct |
8 |
Correct |
9 ms |
6604 KB |
Output is correct |
9 |
Correct |
8 ms |
6092 KB |
Output is correct |
10 |
Correct |
7 ms |
5964 KB |
Output is correct |
11 |
Correct |
6 ms |
5708 KB |
Output is correct |
12 |
Correct |
8 ms |
6416 KB |
Output is correct |
13 |
Correct |
6 ms |
6308 KB |
Output is correct |
14 |
Correct |
8 ms |
5964 KB |
Output is correct |
15 |
Correct |
7 ms |
5964 KB |
Output is correct |
16 |
Correct |
10 ms |
7004 KB |
Output is correct |
17 |
Correct |
11 ms |
6348 KB |
Output is correct |
18 |
Correct |
7 ms |
5708 KB |
Output is correct |
19 |
Correct |
8 ms |
6220 KB |
Output is correct |
20 |
Correct |
7 ms |
5996 KB |
Output is correct |
21 |
Correct |
6 ms |
6092 KB |
Output is correct |
22 |
Correct |
7 ms |
6092 KB |
Output is correct |
23 |
Correct |
6 ms |
5836 KB |
Output is correct |
24 |
Correct |
8 ms |
6296 KB |
Output is correct |
25 |
Correct |
7 ms |
6092 KB |
Output is correct |
26 |
Correct |
7 ms |
6348 KB |
Output is correct |
27 |
Correct |
8 ms |
6220 KB |
Output is correct |
28 |
Correct |
8 ms |
6348 KB |
Output is correct |
29 |
Correct |
7 ms |
6476 KB |
Output is correct |
30 |
Correct |
8 ms |
6476 KB |
Output is correct |
31 |
Correct |
8 ms |
6476 KB |
Output is correct |
32 |
Correct |
9 ms |
6604 KB |
Output is correct |
33 |
Correct |
364 ms |
82684 KB |
Output is correct |
34 |
Correct |
289 ms |
58052 KB |
Output is correct |
35 |
Correct |
348 ms |
79660 KB |
Output is correct |
36 |
Correct |
280 ms |
58052 KB |
Output is correct |
37 |
Correct |
220 ms |
39344 KB |
Output is correct |
38 |
Correct |
200 ms |
33236 KB |
Output is correct |
39 |
Correct |
205 ms |
59084 KB |
Output is correct |
40 |
Correct |
191 ms |
64232 KB |
Output is correct |
41 |
Correct |
147 ms |
64212 KB |
Output is correct |
42 |
Correct |
200 ms |
50180 KB |
Output is correct |
43 |
Correct |
193 ms |
50068 KB |
Output is correct |
44 |
Correct |
329 ms |
114136 KB |
Output is correct |
45 |
Correct |
250 ms |
78384 KB |
Output is correct |
46 |
Correct |
136 ms |
38192 KB |
Output is correct |
47 |
Correct |
327 ms |
59204 KB |
Output is correct |
48 |
Correct |
186 ms |
52408 KB |
Output is correct |
49 |
Correct |
138 ms |
52260 KB |
Output is correct |
50 |
Correct |
215 ms |
53800 KB |
Output is correct |
51 |
Correct |
139 ms |
41376 KB |
Output is correct |
52 |
Correct |
260 ms |
60280 KB |
Output is correct |
53 |
Correct |
161 ms |
53356 KB |
Output is correct |
54 |
Correct |
151 ms |
64408 KB |
Output is correct |
55 |
Correct |
230 ms |
61364 KB |
Output is correct |
56 |
Correct |
218 ms |
67268 KB |
Output is correct |
57 |
Correct |
215 ms |
70208 KB |
Output is correct |
58 |
Correct |
246 ms |
68152 KB |
Output is correct |
59 |
Correct |
241 ms |
68164 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4940 KB |
Output is correct |
2 |
Correct |
3 ms |
4940 KB |
Output is correct |
3 |
Correct |
4 ms |
4940 KB |
Output is correct |
4 |
Correct |
3 ms |
4940 KB |
Output is correct |
5 |
Correct |
11 ms |
6580 KB |
Output is correct |
6 |
Correct |
8 ms |
6092 KB |
Output is correct |
7 |
Correct |
7 ms |
5964 KB |
Output is correct |
8 |
Correct |
9 ms |
6604 KB |
Output is correct |
9 |
Correct |
8 ms |
6092 KB |
Output is correct |
10 |
Correct |
7 ms |
5964 KB |
Output is correct |
11 |
Correct |
6 ms |
5708 KB |
Output is correct |
12 |
Correct |
8 ms |
6416 KB |
Output is correct |
13 |
Correct |
6 ms |
6308 KB |
Output is correct |
14 |
Correct |
8 ms |
5964 KB |
Output is correct |
15 |
Correct |
7 ms |
5964 KB |
Output is correct |
16 |
Correct |
10 ms |
7004 KB |
Output is correct |
17 |
Correct |
11 ms |
6348 KB |
Output is correct |
18 |
Correct |
7 ms |
5708 KB |
Output is correct |
19 |
Correct |
8 ms |
6220 KB |
Output is correct |
20 |
Correct |
7 ms |
5996 KB |
Output is correct |
21 |
Correct |
6 ms |
6092 KB |
Output is correct |
22 |
Correct |
7 ms |
6092 KB |
Output is correct |
23 |
Correct |
6 ms |
5836 KB |
Output is correct |
24 |
Correct |
8 ms |
6296 KB |
Output is correct |
25 |
Correct |
7 ms |
6092 KB |
Output is correct |
26 |
Correct |
7 ms |
6348 KB |
Output is correct |
27 |
Correct |
8 ms |
6220 KB |
Output is correct |
28 |
Correct |
8 ms |
6348 KB |
Output is correct |
29 |
Correct |
7 ms |
6476 KB |
Output is correct |
30 |
Correct |
8 ms |
6476 KB |
Output is correct |
31 |
Correct |
8 ms |
6476 KB |
Output is correct |
32 |
Correct |
9 ms |
6604 KB |
Output is correct |
33 |
Correct |
364 ms |
82684 KB |
Output is correct |
34 |
Correct |
289 ms |
58052 KB |
Output is correct |
35 |
Correct |
348 ms |
79660 KB |
Output is correct |
36 |
Correct |
280 ms |
58052 KB |
Output is correct |
37 |
Correct |
220 ms |
39344 KB |
Output is correct |
38 |
Correct |
200 ms |
33236 KB |
Output is correct |
39 |
Correct |
205 ms |
59084 KB |
Output is correct |
40 |
Correct |
191 ms |
64232 KB |
Output is correct |
41 |
Correct |
147 ms |
64212 KB |
Output is correct |
42 |
Correct |
200 ms |
50180 KB |
Output is correct |
43 |
Correct |
193 ms |
50068 KB |
Output is correct |
44 |
Correct |
329 ms |
114136 KB |
Output is correct |
45 |
Correct |
250 ms |
78384 KB |
Output is correct |
46 |
Correct |
136 ms |
38192 KB |
Output is correct |
47 |
Correct |
327 ms |
59204 KB |
Output is correct |
48 |
Correct |
186 ms |
52408 KB |
Output is correct |
49 |
Correct |
138 ms |
52260 KB |
Output is correct |
50 |
Correct |
215 ms |
53800 KB |
Output is correct |
51 |
Correct |
139 ms |
41376 KB |
Output is correct |
52 |
Correct |
260 ms |
60280 KB |
Output is correct |
53 |
Correct |
161 ms |
53356 KB |
Output is correct |
54 |
Correct |
151 ms |
64408 KB |
Output is correct |
55 |
Correct |
230 ms |
61364 KB |
Output is correct |
56 |
Correct |
218 ms |
67268 KB |
Output is correct |
57 |
Correct |
215 ms |
70208 KB |
Output is correct |
58 |
Correct |
246 ms |
68152 KB |
Output is correct |
59 |
Correct |
241 ms |
68164 KB |
Output is correct |
60 |
Incorrect |
3 ms |
4940 KB |
Output isn't correct |
61 |
Halted |
0 ms |
0 KB |
- |