POJ 1990 MooFest（树状数组）

标签：algorithm

Description

Input

Output

Sample Input

4
3 1
2 5
2 6
4 3

Sample Output

57

先按听力值从小到大排个序，保证当前的牛与前面的联系时，取当前牛的听力值，再用两个树状数组，一个记录所有所有坐标点的位置个数，这样就可以查询大于当前坐标的个数和小于当前坐标的个数，另一个记录所有坐标点的和，这样就可以查询小于和大于当前坐标的所有坐标的和，最后用当前听力值*(大于当前坐标的坐标和—大于当前坐标的个数*当前坐标+小于于当前坐标的个数*当前坐标—小于当前坐标的坐标和)。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=20000+1000;
struct node
{
    int x;
    int y;
}a[maxn];
long long b[maxn];
int c[maxn];
bool cmp(node p,node q)
{
    return p.x<q.x;
}
int low(int k)
{
    return (k&(-k));
}
void update(int k,int v)
{
    while(k<maxn)
    {
        b[k]+=v;
        k+=low(k);
    }
}
void update2(int k,int v)
{
    while(k<maxn)
    {
        c[k]+=v;
        k+=low(k);
    }
}
long long getsum(int k)
{
    long long ans=0;
    while(k>0)
    {
        ans+=b[k];
        k-=low(k);
    }
    return ans;
}
int getsum2(int k)
{
    int ans=0;
    while(k>0)
    {
       ans+=c[k];
       k-=low(k);
    }
    return ans;
}
int main()
{
    long long ans;
    int n;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&a[i].x,&a[i].y);
        }
        ans=0;
        sort(a,a+n,cmp);
        long long temp=0;
        for(int i=0;i<n;i++)
        {
            long long x1=getsum(a[i].y);//小于当前坐标的坐标和
            long long y1=temp-x1;//大于当前坐标的坐标和
            int x2=getsum2(a[i].y);//小于当前坐标的个数
            int y2=i-x2;//大于当前坐标的个数
            ans+=(long long)a[i].x*(y1-a[i].y*y2);
            ans+=(long long)a[i].x*(a[i].y*x2-x1);
            update(a[i].y,a[i].y);
            update2(a[i].y,1);
            temp+=a[i].y;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}

POJ 1990 MooFest（树状数组）

标签：algorithm

原文地址：http://blog.csdn.net/caduca/article/details/40107637