树状数组
如果将两棵树两两配对复杂度O(n^2),超时。考虑成块计算
突破口在Disharmony Value的两个参量计算的差别上。F = abs(D1-D2) S=min(H1,H2) 从式子上看,S容易固定下来,按照下面的方式对树进行配对。
选定一棵树x,让这颗树与高度比它大的树配对,这样S就能固定下来。在这种情况下怎么解决F这一部分。
假设我们知道与x配对的树的数量num,这num棵树的D的和,这num棵树中D小于Dx的数量snum,及它们D的和。那么就能进行整体运算,求出树x与比
它高的树的Disharmony Value和。剩下的工作就水到渠成了,循环n-1次就可以了。
将所有树按高度排序后,建立两个树状数组,一个用来求snum,一个用来求snum棵树D的和。
答案就在问题中。能发现F和S的不同,并想到以S作为突破口,就拿到了解决问题的钥匙
#include<stdio.h>
#include<iostream>
#include<memory.h>
#include<algorithm>
using namespace std;
#define Maxn 100010
int N;
struct node
{
long long x,h,X,H;
}tree[Maxn];
long long ans,Trank[Maxn],Tnum[Maxn],rankMax;
bool cmpH(node a,node b)
{
return a.h<b.h;
}
bool cmpX(node a,node b)
{
return a.x<b.x;
}
//求2^k
int lowbit(int t)
{
return t & ( t ^ ( t - 1 ) );
}
//求前n项和
long long sum(long long T[],int end)
{
long long sum = 0;
while(end > 0)
{
sum += T[end];
end -= lowbit(end);
}
return sum;
}
//增加某个元素的大小
void Add(long long T[],int pos, long long num)
{
while(pos <= rankMax)
{
T[pos] += num;
pos += lowbit(pos);
}
}
int main()
{
while(scanf("%d",&N)!=EOF)
{
ans=0;
memset(tree,0,sizeof(tree));
memset(Trank,0,sizeof(Trank));
memset(Tnum,0,sizeof(Tnum));
for(int i=0;i<N;i++)
scanf("%I64d %I64d",&tree[i].x,&tree[i].h);
sort(tree,tree+N,cmpX);
tree[0].X=1;
for(int i=1;i<N;i++)
{
if(tree[i].x==tree[i-1].x)
tree[i].X=tree[i-1].X;
else
tree[i].X=i+1;
}
rankMax=tree[N-1].X;
sort(tree,tree+N,cmpH);
tree[0].H=1;
for(int i=1;i<N;i++)
{
if(tree[i].h==tree[i-1].h)
tree[i].H=tree[i-1].H;
else
tree[i].H=i+1;
}
Add(Trank,tree[N-1].X,tree[N-1].X);
Add(Tnum,tree[N-1].X,1);
long long snum,rankL,rankM,srank,L,M;
snum=1,srank=tree[N-1].X;
for(int i=N-2;i>=0;i--)
{
rankL=sum(Trank,tree[i].X);
rankM=srank-rankL;
L=sum(Tnum,tree[i].X);
M=snum-L;
ans+=(tree[i].H*(L*tree[i].X-rankL + rankM-M*tree[i].X));
Add(Trank,tree[i].X,tree[i].X);
Add(Tnum,tree[i].X,1);
snum++,srank+=tree[i].X;
}
printf("%I64d\n",ans);
}
return 0;
}