关闭

POJ 1990 MooFest

380人阅读 评论(0) 收藏 举报

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

 


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:206603次
    • 积分:4190
    • 等级:
    • 排名:第7602名
    • 原创:212篇
    • 转载:17篇
    • 译文:0篇
    • 评论:42条
    最新评论