[POI2011] Tree Rotations

一、題目

點此看題

解釋一下這個輸入，如果值爲$0$那麼遞歸地給出它的子樹，否則給出葉子的權值。

二、解法

總的逆序對就等於所有子樹的逆序對之和，而子樹內的選擇對子樹外的選擇沒有影響，這給了我們貪心的可能，考慮換更好還是不換更好，維護一個權值線段樹，在啓發式合併的時候再統計換的個數和不換的個數。

#include <cstdio>
#include <iostream>
using namespace std;
#define ll long long
const int M = 200005; 
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,cnt;ll ans,ans1,ans2;
struct node
{
	int ls,rs,s;
}a[M*30];
void upd(int &x,int l,int r,int p)
{
	if(!x) x=++cnt;
	a[x].s++;
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(mid>=p) upd(a[x].ls,l,mid,p);
	else upd(a[x].rs,mid+1,r,p);
}
void merge(int &x,int y)
{
	if(!x || !y) {x=x+y;return ;}
	a[x].s+=a[y].s;
	ans1+=1ll*a[a[x].ls].s*a[a[y].rs].s;
	ans2+=1ll*a[a[x].rs].s*a[a[y].ls].s;
	merge(a[x].ls,a[y].ls);
	merge(a[x].rs,a[y].rs);
}
void solve(int &x)
{
	int t=read(),ls=0,rs=0;
	if(!t)
	{
		solve(ls);solve(rs);
		ans1=ans2=0;
		x=ls;merge(x,rs);
		ans+=min(ans1,ans2);
	}
	else upd(x,1,n,t);
}
int main()
{
	n=read();
	solve(m);
	printf("%lld\n",ans);
}