POJ1990MooFest(树状数组)

题意：
一群牛参加完牛的节日后都有了不同程度的耳聋，第i头牛听见别的牛声音，别的牛音量必须大于v[i]，当两头牛i，j交流的时候，交流的最小声音为max{v[i],v[j]}*他们之间的距离。现在有n头牛，求他们之间两两交流最少要的音量和。

---

思路：
首先暴力肯定不是不行的O(n^2)。
树状数组的作用是能够快速求出某段区间的和O(logn)，根据这点。我们可以根据牛可以听见声音的音量从小到大进行排序。这样我们只需要考虑在该位置前面有多少个牛的个数和小于该位置牛的坐标和就行了。假设i牛位置前面有xx个小于该位置的牛，他们的位置和为yy。假设前i个牛的坐标总和为ssum,则大于i牛位置的坐标和zz=ssum-yy;因此他们的音量和为P[i].v*((xx*P[i].x-yy)+(zz-(i+1-xx)*P[i].x))。然后累加起来就是总的音量和。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<string>
#include<set>
#include<map>
#include<vector>
using namespace std;
typedef long long int ll;
const int MAX=20000;
struct node
{
    int v,x;
    bool operator <(const node &m)const
    {
        if(v!=m.v)
            return v<m.v;
        return x<m.x;
    }
}P[MAX+10];
ll bit1[MAX+10],bit2[MAX+10];
int lowbit(int i)
{
    return i&-i;
}
void add(long long *bit,int i,int x)
{
    for(int j=i;j<=MAX;j+=lowbit(j))
    {
        bit[j]+=x;
    }
}
long long int sum(ll *bit,int i)
{
    ll ans=0;
    while(i)
    {
        ans+=bit[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    int n;
    scanf("%d",&n);
    memset(bit1,0,sizeof(bit1));
    memset(bit2,0,sizeof(bit2));
    for(int i=0;i<n;i++)
    {
        scanf("%d%d",&P[i].v,&P[i].x);
    }
    sort(P,P+n);
    ll ssum=0;
    ll ans=0;
    for(int i=0;i<n;i++)
    {
        ssum+=P[i].x;
        ll xx=sum(bit1,P[i].x);//小于x的个数
        ll yy=sum(bit2,P[i].x);//小于x的和
        ll zz=ssum-yy;
        add(bit1,P[i].x,1);
        add(bit2,P[i].x,P[i].x);
        ans+=P[i].v*((xx*P[i].x-yy)+(zz-(i+1-xx)*P[i].x));
    }
    printf("%lld\n",ans);
    return 0;
}