Peng Lv

毋意,毋必,毋固,毋我。 言必行,行必果。

导航

POJ 1990 MooFest (树状数组)

题意:由n头牛,不同的听力值v,当i,j想要通话时,需要max(v(i),v(j))*(dist[i]-dist[j])的volume,问这n*(n-1)/2对牛总共的volume时多少。

思路:非常好的一道树状数组,自己想了很长时间没想起来,最后看了结题报告才恍然大悟,对n头牛按v进行排序,由小到大,tre_sumx[i.x]记录前i头牛于x的坐标的总和,tre_cnt[i.x]记录前i头牛坐标低于x的牛的头数,total记录前i头牛的总x和,那么对于每个牛,依靠这三个量就可以计算出i牛和之前的所有牛的volume和。

 

#include <iostream>
#include
<cstdio>
#include
<algorithm>
#include
<memory.h>
#include
<cmath>
#include
<bitset>
#include
<queue>
#include
<vector>
using namespace std;

const int BORDER = (1<<20)-1;
const int MAXSIZE = 37;
const int MAXN = 20100;
const int INF = 1000000000;
#define CLR(x,y) memset(x,y,sizeof(x))
#define ADD(x) x=((x+1)&BORDER)
#define IN(x) scanf("%d",&x)
#define OUT(x) printf("%d\n",x)
#define MIN(m,v) (m)<(v)?(m):(v)
#define MAX(m,v) (m)>(v)?(m):(v)
#define ABS(x) ((x)>0?(x):-(x))

typedef
struct{
int v;
int x;
}Node;

Node node[MAXN];
__int64 tre_cnt[MAXN];
__int64 tre_sumx[MAXN];
int n,n_tre;

bool cmp(const Node& a,const Node& b)
{
if(a.v == b.v)
return a.x < b.x;
return a.v < b.v;
}
int lowbit(int x)
{
return x&(-x);
}
void modify(__int64* tre,const int& ind,const int& delta)
{
for(int i = ind; i <= n_tre; i += lowbit(i))
tre[i]
+= delta;
}
__int64 get_sum(__int64
* tre, const int& ind)
{
__int64 sum
= 0;
for(int i = ind; i > 0; i -= lowbit(i))
sum
+= tre[i];
return sum;
}
int init()
{
CLR(tre_cnt,
0);
CLR(tre_sumx,
0);
n_tre
= 20005;
return 0;
}
int input()
{
for(int i = 0; i < n; ++i)
{
scanf(
"%d%d",&node[i].v,&node[i].x);
}
return 0;
}
int work()
{
int i,j,cnt;
__int64 ans,sum,total;
ans
= 0;
total
= 0;
sort(node,node
+n,cmp);
modify(tre_cnt,node[
0].x,1);
modify(tre_sumx,node[
0].x,node[0].x);
total
+= node[0].x;
for(i = 1; i < n; ++i)
{
sum
= get_sum(tre_sumx,node[i].x);
cnt
= get_sum(tre_cnt,node[i].x);
ans
+= node[i].v*(node[i].x*((cnt<<1)-i)+total-(sum<<1));
modify(tre_cnt,node[i].x,
1);
modify(tre_sumx,node[i].x,node[i].x);
total
+= node[i].x;
}
cout
<<ans<<endl;
return 0;
}
int main()
{
while(IN(n)!=EOF)
{
init();
input();
work();
}
return 0;
}


 

posted on 2010-04-24 10:08  Lvpengms  阅读(683)  评论(0编辑  收藏  举报