poj 1990 MooFest

时间:2022-12-22 14:44:59

题目大意:

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

分析:

首先将这n头牛按照v值从小到大排序(后面说的排在谁的前面,都是基于这个排序)。这样,排在后面的牛和排在前面的牛讲话,两两之间所用的音量必定为后面的牛的v值,这样一来才有优化的余地。然后,对于某头牛i来说,只要关心跟排在他前面的牛交流就好了。我们必须快速地求出排在他前面的牛和他之间距离的绝对值之和ans,只要快速地求出ans,就大功告成。这里需要两个树状数组。

其实只需求出前面小于X的坐标数s2,和那些小于X的坐标之和s1,然后用total记录当前i个x坐标的所有和

距离为:s2*X0 - s1 + total - s1 - x -(i - count)*X;//分两部分来计算,小于X0的,和大于X0的。

因此考虑用两个树状数组来完成计算。

 #include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#include<string>
#include<queue>
#define N 20000
#define LL long long
using namespace std;
struct Node
{
int v,x;
bool operator<(const Node n)const
{
return v<n.v;
}
} a[N+];
LL cnt[N+],s[N+];
int n;
int lowbit(int x)
{
return ((x)&(-(x)));
}
void Update(LL *b,int x,LL val)
{
for(int i=x; i<=N; i+=lowbit(i))
b[i]+=val;
}
LL sum(LL *b,int x)
{
LL ret=;
for(int i=x; i>; i-=lowbit(i))
ret+=b[i];
return ret;
}
int main()
{
//freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF)
{
for(int i=; i<n; i++) scanf("%d%d",&a[i].v,&a[i].x);
sort(a,a+n);
memset(cnt,,sizeof(cnt));
memset(s,,sizeof(s));
LL ans=,total=;
for(int i=; i<n; i++)
{
Update(cnt,a[i].x,);
Update(s,a[i].x,a[i].x);
total+=a[i].x;
LL s1=sum(cnt,a[i].x-);
LL s2=sum(s,a[i].x-);
//printf("%I64d %I64d====\n",s1*a[i].x-s2, total-s2-a[i].x-a[i].x*(i-s1));
ans+=a[i].v*(s1*a[i].x-s2 +total-s2-a[i].x-a[i].x*(i-s1));
}
printf("%lld\n",ans);
}
return ;
}