bzoj 2212(線段樹合併)
阿新 • • 發佈:2019-02-08
傳送門
題解:權值線段樹從下往上合併,每個點統計兩個兒子中某一個交換/不交換的答案,取min後加到總答案中。不禁讓人想起CDQZ Challenge 13
P.S.形態樹開兩倍空間,因為這種讀入方式肯定會讀成一棵滿二叉樹(不管存不存在子節點先加了時間戳)。另外,定義變數要搞清楚哪個是輸入的形態樹,哪個是線段樹。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+4;
int st,n,a[MAXN<<1],son[2][MAXN<<1],tim=0 ,tot=0;
int root[MAXN<<1],lc[MAXN*19],rc[MAXN*19],siz[MAXN*19];
ll t1,t2,ans=0;
inline int read() {
int x=0;char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x;
}
void init(int &x) {
x=++tim;
a[x ]=read();
if (!a[x]) init(son[0][x]),init(son[1][x]);
}
inline void pushup(int rt) {
siz[rt]=siz[lc[rt]]+siz[rc[rt]];
}
void insert(int &rt,int l,int r,int val) {
if (!rt) rt=++tot;
if (l==r) {siz[rt]=1;return ;}
int mid=(l+r)>>1;
if (val<=mid) insert(lc[rt],l,mid,val);
else insert(rc[rt],mid+1,r,val);
pushup(rt);
}
int merge(int x,int y) {
if (!x) return y;
if (!y) return x;
t1+=(ll)siz[lc[x]]*siz[rc[y]];
t2+=(ll)siz[rc[x]]*siz[lc[y]];
lc[x]=merge(lc[x],lc[y]);
rc[x]=merge(rc[x],rc[y]);
pushup(x);
return x;
}
void work(int x) {
if (a[x]) return ;
work(son[0][x]),work(son[1][x]);
t1=t2=0;
root[x]=merge(root[son[0][x]],root[son[1][x]]);
ans+=min(t1,t2);
}
int main() {
// freopen("bzoj 2212.in","r",stdin);
n=read();
init(st);
for (register int i=1;i<=tim;++i)
if (a[i]) insert(root[i],1,n,a[i]);
work(st);
printf("%lld\n",ans);
return 0;
}