POJ 1990 MooFest

http://poj.org/problem?id=1990

题意：FJ有n头牛，排列成一条直线(不会在同一个点)，给出每头牛在直线上的坐标x。另外，每头牛还有一个自己的声调v，如果两头牛(i和j)之间想要沟通的话，它们必须用同个音调max(v[i],v[j])，沟通起来消耗的能量为：max(v[i],v[j]) * 它们之间的距离。问要使所有的牛之间都能沟通(两两之间)，总共需要消耗多少能量。

思路：树状数组，很好的一道题 。先按照v的大小排序（如果两头牛的v相同，可以任意排序），然后对于没头牛（d，v），维护两个树状数组c1[ ]，c2[ ]，其中c1[]存放x小于d的点的个数，c2[]存放x比d小的牛的x的和。

对于第i头牛，（d，v）的value为：v * ( d * c1[d - 1] -c2[d - 1]  + (sum[ i -1 ] - c2[d-1]) - (i-1-d) * c1[d-1] ) ;

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define MAXN 20010
struct Node{
	long long d,v ;	
}p[MAXN] ;
bool comp(const Node& n1, const Node& n2){
	return n1.v < n2.v ;	
}
long long N ,MAX_DIS;
long long c1[MAXN] ;
long long c2[MAXN] ;
long long sum[MAXN] ;
long long lowbit(long long i){
	return (i & (i ^ (i - 1))) ;	
}
long long sum1(long long i){
	long long res = 0 ;
	while(i>=1){
		res += c1[i] ;
		i -= lowbit(i) ;	
	}
	return res ;
}
long long sum2(long long i){
	long long res = 0 ;
	while(i >= 1){
		res += c2[i] ;
		i -= lowbit(i) ;	
	}	
	return res ;
}
void add1(long long i){
	while(i <= MAX_DIS){
		c1[i]++ ;
		i += lowbit(i);	
	}		
}
void add2(long long i, long long ad){
	while(i <= MAX_DIS){
		c2[i] += ad ;
		i += lowbit(i) ;	
	}	
}
int main(){
	long long d , v ;
	while(scanf("%lld",&N) == 1){
		MAX_DIS = -1 ;
		for(long long i=1;i<=N;i++){
			scanf("%lld %lld",&p[i].v , &p[i].d);	
			MAX_DIS = (MAX_DIS > p[i].d ? MAX_DIS :p[i].d) ;
		}	
		sort(p+1,p+1+N,comp);
		memset(sum , 0 ,sizeof(sum));
		for(long long i=1;i<=N;i++){
			sum[i] = sum[i-1] +  p[i].d ;
		}
		memset(c1,0,sizeof(c1));
		memset(c2,0,sizeof(c2));		
		long long ans = 0 ;
		for(long long i=1;i<=N;i++){
			d = p[i].d ;
			v = p[i].v ;
			long long n1 = sum1(d-1);	
			long long s1 = sum2(d-1);	
			add1(d);
			add2(d,d);
			ans += v*(d*n1-s1+sum[i-1]-s1-(i-1-n1)*d);
			//printf("%d %d %d %d %d\n",n1,s1,d,v,ans);
		}
		printf("%lld\n",ans);
	}
	return 0 ;	
}