POJ1990 - MooFest

题目大意

给定N对二元组（x1,y1),(x2,y2),(x3,y3)…(xn,yn).每两个二元组之间(假设分别为i和j,i!=j)将会产生一个值ans(i,j)=max(xi,xj)*|yi-yj|,计算出sigma(ans(i,j))(1<=i,j<=n,i!=j)

题解

我们可以先对x值按升序排序，之后就是用两个树状数组进行维护了，公式推导过程不知道怎么讲。。。直接看代码好了。。。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#define MAXN 20005

using namespace std;

typedef struct

{

    int v;

    int x;

} NODE;

NODE a[MAXN];

int n;

long long c[MAXN],b[MAXN];

int lowbit(int x)

{

    return x&-x;

}

int sum(long long p[],int x)

{

    int ret=0;

    while(x>0)

    {

        ret+=p[x];

        x-=lowbit(x);

    }

    return ret;

}

void add(long long p[],int x,int d)

{

    while(x<=MAXN)

    {

        p[x]+=d;

        x+=lowbit(x);

    }

}

bool cmp(NODE a,NODE b)

{

    return a.v<b.v;

}

int main(void)

{

    long long ans=0,total=0,cnt,sums;

    int i;

    scanf("%d",&n);

    for(i=1; i<=n; i++)

        scanf("%d%d",&a[i].v,&a[i].x);

    sort(a+1,a+n+1,cmp);

    memset(c,0,sizeof(c));

    memset(b,0,sizeof(b));

    for(i=1; i<=n; i++)

    {

        cnt=sum(b,a[i].x);

        sums=sum(c,a[i].x);

        ans+=a[i].v*(total-2*cnt+(2*sums-i+1)*a[i].x);

        add(b,a[i].x,a[i].x);

        add(c,a[i].x,1);

        total+=a[i].x;

    }

    cout<<ans<<endl;

    return 0;

}