poj 1990 MooFest

题意：

若干头牛排列在一条坐标轴上（位置都不同），每头牛都有一个音量vi，任意两头牛i,j之间要交流，他们发出的音量就是max(vi,vj) * 它们的距离。

问n / (n-1)对牛交流的音量的总和。

思路：

先无脑n ^ 2了一波，果然tle了，是我太naive。

首先把牛按照音量递增排序，对于当前的牛，只需要知道数组中在它前面的牛到它的距离的绝对值之和，那么就很好办了。

关键就是求前面的牛到它的距离的绝对值之和。

用两个树状数组，一个维护数量的前缀，一个维护位置和的前缀。

用一个dis来累加当前的距离的总和。

对于当前的牛，数量树状数组可以求出小于它的位置的牛的个数cnt，位置和树状数组可以求出小于等于它的牛的位置之和sum。

那么它前面的小于它的位置的牛到它的距离之和就是cnt * x[i] - sum，

大于它的位置的牛到它的距离之和就是dis - sum - (i - cnt) * x[i]。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 2e5 + 10;
int c[N];
long long d[N];
struct node
{
    int v,p;
} a[N];
int mx;
int lowbit(int x)
{
    return x&(-x);
}
void addx(int x)
{
    for (int i = x;i <= 20000;i += lowbit(i)) c[i]++;
}
int getx(int x)
{
    int ans = 0;
    for (int i = x;i > 0;i -= lowbit(i)) ans += c[i];
    return ans;
}
void addd(int x,int y)
{
    for (int i = x;i <= 20000;i += lowbit(i)) d[i] += y;
}
long long getd(int x)
{
    long long ans = 0;
    for (int i = x;i > 0;i -= lowbit(i)) ans += d[i];
    return ans;
}
bool cmp(node aa,node bb)
{
    if (aa.v == bb.v) return aa.p < bb.p;
    return aa.v < bb.v;
}
int main()
{
    int n;
    while (scanf("%d",&n) != EOF)
    {
        memset(c,0,sizeof(c));
        memset(d,0,sizeof(d));
        long long ans = 0;
        for (int i = 0;i < n;i++) scanf("%d%d",&a[i].v,&a[i].p);
        sort(a,a+n,cmp);
        long long dis = 0;
        for (int i = 0;i < n;i++)
        {
            int cnt = getx(a[i].p);
            long long tmp = 0;
            long long tmpd = getd(a[i].p);
            tmp += 1LL * cnt * a[i].p - tmpd;
            tmp += dis - tmpd - 1LL * (i - cnt) * a[i].p;
            ans += tmp * a[i].v;
            addx(a[i].p);
            addd(a[i].p,a[i].p);
            dis += a[i].p;
        }
        printf("%lld
",ans);
    }
    return 0;
}