POJ 1990 MooFest【 树状数组 】

时间:2021-08-22 14:46:22

题意:给出n头牛,每头牛有一个听力v,坐标x,两头牛之间的能量为max(v1,v2)*dist(v1,v2),求总的能量值

先将每头牛按照v排序,排完顺序之后,会发现有坐标比当前的x小的,会有坐标比当前的x大的

假设坐标比x小的有num个

那么

距离之和 = x*num - 前面坐标的和 + 后面坐标的和 - (n-num-1)* x

又因为

后面坐标的和 = 整个区间坐标的和 - 前面坐标的和

所以用两个数组,一个数组用来算个数,另外一个数组用来算和

学习的这一篇题解--http://www.acmtime.com/?p=403

 #include<iostream>
#include<cstdio>
#include<cstring>
#include <cmath>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<algorithm>
using namespace std; typedef long long LL;
const int INF = (<<)-;
const int mod=;
const int maxn=; LL a[maxn];//这个是用来求比x小的个数的
LL b[maxn];//这个是用来求前面的和
LL c[maxn];
int n; struct node{
int x,v;
}q[maxn]; int cmp(node n1,node n2){
return n1.v < n2.v;
} int lowbit(int x){ return x & (-x);} LL sum(LL c[],int x){
LL ret=;
while( x >){
ret += c[x]; x-=lowbit(x);
}
return ret;
} void add( LL c[],int x,int d){
while(x < maxn){
c[x]+=d;x+=lowbit(x);
}
} int main(){
scanf("%d",&n);
for(int i=;i<=n;i++) scanf("%d %d",&q[i].v,&q[i].x);
sort(q+,q+n+,cmp); memset(a,,sizeof(a));
memset(b,,sizeof(b));
LL cnt =;
for(int i=;i<=n;i++){
int x = sum(a,q[i].x); //统计坐标比x小的有多少个
LL alltotal = sum(b,maxn);//统计这个区间所有坐标的和
LL total = sum(b,q[i].x);//统计前面坐标比x小的的和
cnt += q[i].v * (x * q[i].x - total + alltotal - total - ( i - x -) * q[i].x);
add(a,q[i].x,);
add(b,q[i].x,q[i].x);
}
printf("%I64d\n",cnt);
return ;
}

加油-------------

goooooooooooooo-------- -------