原文链接http://www.cnblogs.com/zhouzhendong/p/8079786.html
题目传送门 - BZOJ3286
题意概括
给一棵n(1≤n≤200000个叶子的二叉树,可以交换每个点的左右子树,要求前序遍历叶子的逆序对最少。
题解
线段树合并。
博主很懒,题解不写了。
这份代码是仿照别人的写的。
代码
#include <cstring> #include <cstdio> #include <cmath> #include <algorithm> #include <cstdlib> using namespace std; typedef long long LL; const int N=200005; int n,val[N*2],son[N*2][2],tot=0,rt; int root[N*2],ls[N*2*20],rs[N*2*20]; LL sum[N*2*20]; void dfsread(int &x){ x=++tot; scanf("%d",&val[x]); if (!val[x]){ dfsread(son[x][0]); dfsread(son[x][1]); } } void pushup(int rt){ sum[rt]=sum[ls[rt]]+sum[rs[rt]]; } LL sumL,sumR; void add(int &rt,int L,int R,int v){ rt=++tot; int mid=(L+R)>>1; if (L==R){ sum[rt]=1; return; } if (v<=mid) add(ls[rt],L,mid,v); else add(rs[rt],mid+1,R,v); pushup(rt); } int merge(int a,int b){ if (!a||!b) return a+b; sumL+=sum[rs[a]]*sum[ls[b]]; sumR+=sum[ls[a]]*sum[rs[b]]; ls[a]=merge(ls[a],ls[b]); rs[a]=merge(rs[a],rs[b]); pushup(a); return a; } LL dfs(int x){ LL ans=0; if (!val[x]){ ans=dfs(son[x][0])+dfs(son[x][1]); sumL=sumR=0; root[x]=merge(root[son[x][0]],root[son[x][1]]); ans+=min(sumL,sumR); } else add(root[x],1,n,val[x]); return ans; } int main(){ scanf("%d",&n); dfsread(rt); printf("%lld",dfs(rt)); return 0; }
时间: 2024-10-16 12:27:06