bzoj 4240: 有趣的家庭菜园 树状数组

       （随手写了一发rank1什么鬼。。）

       因为4239有点繁琐，就先跳过去了。。。然后看完这道题目就懵逼了。。o(╯□╰)o

       但是仔细想想会发现对于一棵草，它的移动是不会影响到比它更高的草的，因此我们可以从小到大移动草，并且贪心地移动到较小的一边，换句话说令f(i,j,k)表示i~j中比k大的数的个数，那么对于某一棵草(x,y)（位置为x高度为y）的答案为min{f(1,x-1,y),f(x+1,n,y)}。那么可以从大到小添入节点然后用树状数组维护即可。

AC代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int n,c[300005]; struct node{ int h,id; }a[300005];
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
bool cmp(node x,node y){ return x.h>y.h; }
void add(int x){
	for (; x<=n; x+=x&-x) c[x]++;
}
int qry(int x){
	int t=0; for (; x; x-=x&-x) t+=c[x]; return t;
}
int main(){
	n=read(); int i;
	for (i=1; i<=n; i++){
		a[i].h=read(); a[i].id=i;
	}
	sort(a+1,a+n+1,cmp);
	int head=1; long long ans=0;
	for (i=1; i<=n; i++){
		if (a[i].h!=a[i-1].h)
			while (head<i) add(a[head++].id);
		int tmp=qry(a[i].id); ans+=min(tmp,head-1-tmp);
	}
	printf("%lld\n",ans);
	return 0;
}

by lych

2016.3.19