hdu3015 Disharmony Trees(树状数组）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3015

题意：在路边有一行树，给出它们的坐标和高度，先按X坐标排序。记录排名，记为rankx，再按它们的高度排序，记录排名，记为rankh。

两颗树i,j的差异度为fabs(rankx[i]-rankx[j])*min(rankh[i],rankh[j]),最后求出任异两颗树差异度的和。（注意：有并列排名时算它们的实际排名，如1，2，2，4，没有第三名）。

解析：差异度等于F*S，突破点就在这个S上，在按高度排序以后，我们可以从低到高枚举每一颗树，当前树i和比它高的树匹配，这样S就是确定的了，就是当前这颗树的rankh，现在考虑F，F = fabs(rankx[i]-rankx[j])，假设我们知道i点以后有多少颗树的坐标大于第i颗树，记为cntb[i]，并记录它们的rankx的和，记为sumb[i]；我们可以利用cntb[i]算出高于第i颗树所有树中坐标小于rankx[i]的树的个数cnts[i]，同理计算出它们的和sums[i]。这样就可以算出第i颗树与所有高于第i树的差异度之和，

等于rankh[i] * (sumb[i] - cntb[i]*rankx[i] + cnts[i]*rankx[i] - sums[i]); cntb与sumb数组可以用两颗树状数组用求逆序数的思想来得出，这里就不再多说了。

参考代码：

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

#define M 100005
#define clr(arr,v) memset(arr,v,sizeof(arr))
typedef __int64 I64;

struct point{
	int x,h;
	int rankx,rankh;
}data[M],data1[M];

I64 cntb[M],sum[M],sumb[M];
I64 tree1[M],tree2[M];

bool cmp1(point po1,point po2) { return po1.h < po2.h; }
bool cmp2(point po1,point po2) { return po1.x < po2.x; }

int lowbit(int x){ return x&-x; }
int get_num(I64 tree[],int x,int n)
{
	if(x >= n) return tree[n];
	return tree[x]+get_num(tree,x+lowbit(x),n);
}

void update(I64 tree[],int x,int value)
{
	if(x <= 0) return ;
	tree[x] += value;
	update(tree,x - lowbit(x),value);
}

void init(int n)
{
	clr(tree1,0);
	clr(tree2,0);
	clr(cntb,0);
	clr(sum,0); 
	clr(sumb,0);
	int rankx = 0;
	sort(data,data+n,cmp2);
	for(int i = 0;i < n;++i)
	{
		if(i == 0 || data[i].x != data[i-1].x)
			data[i].rankx = ++rankx;
		else 
			data[i].rankx = data[i-1].rankx,++rankx;
	}
	int rankh = 0;
	sort(data,data+n,cmp1);
	for(int i = 0;i < n;++i)
	{
		if(i == 0 || data[i].h != data[i-1].h)
			data[i].rankh = ++rankh;
		else
			data[i].rankh = data[i-1].rankh,++rankh;
	}
	for(int i = n-1;i >= 0;--i)
	{
		sum[i] = data[i].rankx + sum[i+1];
		cntb[i] += get_num(tree1,data[i].rankx,rankx);
		update(tree1,data[i].rankx-1,1);
		sumb[i] = get_num(tree2,data[i].rankx,rankx);
		update(tree2,data[i].rankx-1,data[i].rankx);
	}
}

I64 solve(int n) 
{
	I64 res = 0;
	for(int i = 0;i < n;++i)
		res += data[i].rankh * (sumb[i] - data[i].rankx*cntb[i] + data[i].rankx * (n-1-i-cntb[i]) - (sum[i] - sumb[i] - data[i].rankx));
	return res;
}

int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		for(int i = 0;i < n;++i)
			scanf("%d%d",&data[i].x,&data[i].h);
		init(n);
		printf("%I64d\n",solve(n));
	}
	return 0;
}