POJ - 1990 MooFest 树状数组

  POJ - 1990 MooFest

题意：

给定 n 头牛的 听力阈值 和 所在位置，第 i 头牛和第 j 头牛相互交流所需的最小值为 max(Vi，Vj)*(Xi-Xj)。求所有牛相互交流的总和。

思路：

将牛的听力阈值降序排列。计算时当前牛的听力阈值即为最大所以只需求出 距离和牛数 就能算出当前牛到其他已经存在树状数组中牛的的总和，并更新。

这里用到两个树状数组

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=20000;
ll c[maxn+10][2];  // 0  距离
struct node        //  1 牛数
{
    int v,x;
}p[maxn+10];
bool cmp(node a,node b)
{
    if(a.v==b.v)
        return a.x<b.x;
    return a.v<b.v;
}
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int v,int f)
{
    while(x<=maxn)
    {
        c[x][f] += v;
        x += lowbit(x);
    }
}
ll sum(int x,int f)
{
    ll ans=0;
    while(x)
    {
        ans += c[x][f];
        x -= lowbit(x);
    }
    return ans;
}
int main()
{
    int n;
    scanf("%d",&n);
    memset(c,0,sizeof(c));
    for(int i=1;i<=n;i++)
        scanf("%d%d",&p[i].v,&p[i].x);
    sort(p+1,p+n+1,cmp);
    ll a1,a2,b1,b2,ans=0;
    for(int i=1;i<=n;i++)
    {
        a1 = sum(p[i].x,0);   // 比p[i].x 距离小的牛的距离和
        a2 = sum(maxn,0)-a1;   // 比p[i].x 距离大的牛的距离和
        b1 = sum(p[i].x,1);   //在p[i].x 前边的牛数
        b2 = sum(maxn,1)-b1;    //  在p[i].x后边的牛数
        ans += (b1*p[i].x-a1+a2-b2*p[i].x)*p[i].v;
        add(p[i].x,p[i].x,0);  //把距离存入树状数组
        add(p[i].x,1,1);       // 牛数加 1
    }
    printf("%lld\n",ans);
    return 0;
}