poj 1990 MooFest 树状数组

第三道树状数组模板题，但还是看了题解才会，好菜啊。。。。

首先，我们要确定这道题的基本思路，那就是对每一个点（x0，v0），找到对应的其他点（其v值小于等于v0），并且算出其贡献值。

看到小于等于，脑子里立马浮现出树状数组！但是这道题比较麻烦，关键在于它的贡献算法比较奇葩，居然还要算x差的绝对值。那么怎么做呢？

关键就在于树状数组存的是什么东西（的和）！一般来说，按照套路，我们都是存x值的。这样的话，我们可以算出x的和。

但是够吗？不够！因为题目里要算x差的绝对值，我们不妨把v值小于v0的点分为两类，一类的x小于x0，一类的大于等于x0.最后可以得到

ans+=(cnt*a[i].x-sum+(total_sum-sum)-(i-cnt-1)*a[i].x)*a[i].v;

其中cnt为x值小于x0的点的个数，total_sum为目前所有点x值的总和,sum为所有x值小于x0的点的x值的总和。

sum可以通过我们上面说的那个树状数组得到，而cnt就需要再建造一个树状数组了，这个树状数组里存的是x出现的次数。

下面贴代码

#include<cstdio>
#include<cstring>
#include<iso646.h>
#include<iostream>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
const long long maxn=40000;
struct myk
{
    long long v;
    long long x;
};
myk a[maxn];
long long n,tree1[maxn],tree2[maxn];//tree1 记录x出现的总次数，tree2记录x的和
bool cmp(myk aa,myk bb)
{
    return(aa.v<bb.v);
}
void add(long long tree[],long long k,long long num)
{
    while(k<maxn)
    {
        tree[k]+=num;
        k+=k&(-k);
    }
}
long long read(long long tree[],long long k)
{
    long long sum=0;
    while(k)
    {
        sum+=tree[k];
        k-=k&(-k);
    }
    return sum;
}
int main(void)
{
    long long sum=0,total_sum=0,cnt=0,ans=0;
    cin>>n;
    memset(tree1,0,sizeof(tree1));
    memset(tree2,0,sizeof(tree2));
    for(int i=1;i<=n;i++)
        cin>>a[i].v>>a[i].x;
    sort(a+1,a+n+1,cmp);
    //printf("At first ans=%lld\n",ans);
    for(int i=1;i<=n;i++)
    {
        sum=read(tree2,a[i].x);
        //printf("1:Now i=%d and ans=%I64d\n",i,ans);
        cnt=read(tree1,a[i].x);
        //printf("2:Now i=%d and ans=%lld\n",i,ans);
        ans+=(cnt*a[i].x-sum+(total_sum-sum)-(i-cnt-1)*a[i].x)*a[i].v;
        //printf("3:Now i=%d and ans=%lld\n",i,ans);
        total_sum+=a[i].x;
        add(tree1,a[i].x,1);
        add(tree2,a[i].x,a[i].x);
    }
    cout<<ans;
    return 0;
}