POJ 1990 MooFest --树状数组

题意：牛的听力为v，两头牛i,j之间交流，需要max(v[i],v[j])*dist(i,j)的音量。求所有两两头牛交谈时音量总和∑(max(v[i],v[j])*abs(x[j]-x[i])) ，x[i]表示i的坐标

解法：将牛按听力从小到大排序，这样就可以使每次算一头牛与别的牛的值时听力永远取自己的听力。

建两棵树状数组，一颗维护牛的坐标，一颗维护牛的个数。

当遍历到某头牛 i 时，求出它的左边有多少头牛，记为L，右边记为R，并且算出左边的坐标之和Lsum,右边坐标值的和Rsum。

那么此时 总的和要加上 (L*x[i]-Lsum)*V[i] （左边） +  (Rsum-R*x[i])*V[i] （右边）， 仔细想想就知道了。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 20007

int c[N],cnt[N],n,maxi;
struct node
{
    int val,x;
}p[N];

int cmp(node ka,node kb) { return ka.val < kb.val; }
int lowbit(int x) { return x&-x; }

void modify(int *c,int x,int val)
{
    while(x <= maxi)
        c[x] += val, x += lowbit(x);
}

int getsum(int *c,int x)
{
    int res = 0;
    while(x > 0) { res += c[x]; x -= lowbit(x); }
    return res;
}

int main()
{
    int i,j;
    while(scanf("%d",&n)!=EOF)
    {
        maxi = 0;
        for(i=1;i<=n;i++)
            scanf("%d%d",&p[i].val,&p[i].x), maxi = max(maxi,p[i].x);
        sort(p+1,p+n+1,cmp);
        memset(c,0,sizeof(c));
        memset(cnt,0,sizeof(cnt));
        lll sum = 0;
        for(i=1;i<=n;i++)
        {
            int L = getsum(cnt,p[i].x);
            int R = getsum(cnt,maxi) - getsum(cnt,p[i].x-1);
            int Lsum = getsum(c,p[i].x);
            int Rsum = getsum(c,maxi) - getsum(c,p[i].x-1);
            sum += 1LL*(p[i].x*L-Lsum)*p[i].val;
            sum += 1LL*(Rsum-p[i].x*R)*p[i].val;
            modify(c,p[i].x,p[i].x);
            modify(cnt,p[i].x,1);
        }
        cout<<sum<<endl;
    }
    return 0;
}