题解 BZOJ 2212 [Poi2011]Tree Rotations

学习 http://www.docin.com/p-598829500.html 线段树合并中的一道例题。
详细见那个课件。
复杂度$O(nlgn)$.
需要注意下空间

code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
int n;
int LS[400005],RS[400005];
int cc;
int Root;
int w[400005];
int siz[400005];
int ls[15000500],rs[15000500],sum[15000500];
int num;
int stk[15000500],top;
int getnew(){
if(top>0){
ls[stk[top-1]]=rs[stk[top-1]]=sum[stk[top-1]]=0;
return stk[--top];
}
return ++num;
}
void add(int &o,int l,int r,int pos){
if(!o) o=getnew();
if(l==r){
sum[o]=1;
}else{
int mid=(l+r)>>1;
if(pos<=mid) add(ls[o],l,mid,pos);else add(rs[o],mid+1,r,pos);
sum[o]=1;
}
}
int read(){
int x;
scanf("%d",&x);
int now=++cc;
if(!x){
LS[now]=read();
RS[now]=read();
siz[now]=siz[LS[now]]+siz[RS[now]]+1;
}else{
add(w[now],1,n,x);
siz[now]=1;
}
return now;
}
LL ans;
LL ans0,ans1;
int merge(int a,int b,int l,int r){
if(!a) return b;
if(!b) return a;
int now=getnew();
int mid=(l+r)>>1;
int lft=sum[ls[a]]+sum[ls[b]];
int rgt=sum[rs[a]]+sum[rs[b]];
ans0+=sum[rs[a]]*1LL*sum[ls[b]];
ans1+=sum[ls[a]]*1LL*sum[rs[b]];
if(lft<rgt){
rs[now]=merge(rs[a],rs[b],mid+1,r);
ls[now]=merge(ls[a],ls[b],l,mid);
}else{
ls[now]=merge(ls[a],ls[b],l,mid);
rs[now]=merge(rs[a],rs[b],mid+1,r);
}
stk[top++]=a;
stk[top++]=b;
sum[now]=sum[ls[now]]+sum[rs[now]];
return now;
}
void dfs(int x){
if(!LS[x]) return;
else{
if(siz[LS[x]]<siz[RS[x]]) dfs(RS[x]),dfs(LS[x]);
else dfs(LS[x]),dfs(RS[x]);
ans0=ans1=0;
w[x]=merge(w[LS[x]],w[RS[x]],1,n);
ans+=min(ans0,ans1);
}
}
int main(){
http://www.docin.com/p-598829500.html
scanf("%d",&n);
Root=read();
dfs(1);
printf("%lld\n",ans);
return 0;
}

评论小助手

评论正在加载...

Tip: 点击下方链接切换到 Disqus 评论框可以获得邮件提醒哦
🗣️ 加载 Disqus 评论框