#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int sz =2e5+1;
const int ssz=5e3+1;
const int mod=1e9+7;
int p,n,q,timer,m,cnt;
long long lb,rb;
queue<int>qe;
int ans[sz];
int lf[sz],rt[sz];
int lc[sz*40],rc[sz*40];
int cap[sz*40];
int val[sz];
int fa[sz];
void read(int &now){
now = ++cnt;
scanf("%d",&val[now]);
if(!val[now]){
read(lf[now]);
read(rt[now]);
}
}
void upd(int pos,int &num,int l = 1, int r = n){
if(!num) num = ++timer;
cap[num] = 1;
if(l == r) return;
int mid = (l + r)>>1;
if(pos <= mid) upd(pos,lc[num],l,mid);
else upd(pos,rc[num],mid + 1, r);
}
int merge(int roota, int rootb){
if(!roota or !rootb) return roota + rootb;
cap[roota]+=cap[rootb];
lb += cap[rc[rootb]] * cap[lc[roota]];
rb += cap[rc[roota]] * cap[lc[rootb]];
lc[roota] = merge(lc[roota],lc[rootb]);
rc[roota] = merge(rc[roota],rc[rootb]);
rootb = 0;
return roota;
}
long long dfs(int u){
long long res = 0;
if(!val[u]){
res = dfs(lf[u]) + dfs(rt[u]);
lb = rb = 0;
fa[u] = merge(fa[lf[u]],fa[rt[u]]);
res = res + min(lb,rb);
}
else upd(val[u],fa[u]);
return res;
}
int main(){
scanf("%d",&n);
int root;
read(root);
long long res = dfs(root);
cout<<res<<endl;
}
Compilation message
rot.cpp: In function 'void read(int&)':
rot.cpp:18:11: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
18 | scanf("%d",&val[now]);
| ~~~~~^~~~~~~~~~~~~~~~
rot.cpp: In function 'int main()':
rot.cpp:54:12: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
54 | scanf("%d",&n);
| ~~~~~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
0 ms |
340 KB |
Output is correct |
5 |
Correct |
0 ms |
304 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
288 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
468 KB |
Output is correct |
2 |
Correct |
1 ms |
444 KB |
Output is correct |
3 |
Correct |
1 ms |
440 KB |
Output is correct |
4 |
Correct |
1 ms |
468 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
1108 KB |
Output is correct |
2 |
Correct |
3 ms |
1108 KB |
Output is correct |
3 |
Correct |
2 ms |
1216 KB |
Output is correct |
4 |
Correct |
3 ms |
1236 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
10 ms |
3028 KB |
Output is correct |
2 |
Correct |
7 ms |
2880 KB |
Output is correct |
3 |
Correct |
18 ms |
7124 KB |
Output is correct |
4 |
Correct |
7 ms |
3412 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
33 ms |
12248 KB |
Output is correct |
2 |
Correct |
26 ms |
12244 KB |
Output is correct |
3 |
Correct |
40 ms |
14284 KB |
Output is correct |
4 |
Correct |
33 ms |
15156 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
67 ms |
23352 KB |
Output is correct |
2 |
Correct |
49 ms |
22692 KB |
Output is correct |
3 |
Correct |
58 ms |
21452 KB |
Output is correct |
4 |
Correct |
51 ms |
21836 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
21 ms |
3904 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
1083 ms |
48680 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
68 ms |
65536 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
95 ms |
65536 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |