1. 程式人生 > >bzoj 2212(線段樹合併)

bzoj 2212(線段樹合併)

傳送門
題解:權值線段樹從下往上合併,每個點統計兩個兒子中某一個交換/不交換的答案,取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; }