题目链接: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;
}