【poj 1990】MooFest（树状数组）

传送门biu~

题目大意：一群牛站成一条直线，第i头牛听见别人的讲话，别人的音量必须大于v[i]。两头牛i、j必须用同一个声音max(v[i],v[j])交流。消耗的能量为 声音大小*他们之间的距离。现在有n头牛，给定n头牛的v[i]和坐标，求他们之间两两交流所需要的能量和。

思路：把每头牛按v值大小排序。逐一加入时，第i头牛与第1~(i-1)头牛交流的声音大小一定是v[i]。这样就用树状数组numtree维护有多少头牛，用loctree维护牛的坐标和，用tot维护当前已加入的所有的牛的坐标和。

所以对于牛i，他前面有num头牛，这些牛的坐标和为loc，第i头牛的坐标为xi，声音v值为vi。那么它和比它坐标小的牛之间的坐标差front=num*xi-loc；它和比他坐标大的牛之间的坐标差back=(tot-loc-xi)-xi(i-num-1)。

最后让ans的值加上v[i]*(front+back)就可以了。

#include<bits/stdc++.h>
#define lowbit(x) (x&(-x))
#define MAXN 20005
#define ll long long
using namespace std;
struct COW{
int v,x;
}a[MAXN];
inline bool cmp(COW a,COW b){return a.v<b.v;}
ll numtree[MAXN],loctree[MAXN],n;
int Nmax;
inline void add(ll *a,int x,int num)
{
while(x<=Nmax)
{
a[x]+=num;
x+=lowbit(x);
}
}
inline ll search(ll *a,int x)
{
ll re=0;
while(x)
{
re+=a[x];
x-=lowbit(x);
}
return re;
}
int main()
{
while(~scanf("%d",&n))
{
ll ans=0,tot=0;
memset(numtree,0,sizeof(numtree));
memset(loctree,0,sizeof(loctree));
Nmax=0;
for(int i=1;i<=n;++i)
{
scanf("%d%d",&a[i].v,&a[i].x);
Nmax=max(a[i].x,Nmax);
}
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;++i)
{
add(numtree,a[i].x,1);
add(loctree,a[i].x,a[i].x);
tot+=a[i].x;
ll num=search(numtree,a[i].x-1);
ll loc=search(loctree,a[i].x-1);
ll front=num*a[i].x-loc;
ll back=(tot-loc-a[i].x)-a[i].x*(i-num-1);
ans+=a[i].v*(front+back);
}
printf("%lld\n",ans);
}
return 0;
}