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 ;
}