poj1990 MooFest

x轴上有n个坐标点，每个点x[i]有一个权值v[i]，距离定义为：dis[i][j]=max{v[i],v[j]}*abs(x[i]-x[j])，求n个点所有距离之和。

做法：按v[i]从小到大排序，dp[i]定义为以1-i点之间以i结尾的点对距离之和。dp[i]=v[i]*sum{ abs(x[i]-x[j]),1<=j<=i-1 }。对于x[i]的系数，只要知道x数组1 - i-1位中比x[i]小的个数，记为c[i]，x[j]的正负只要知道x数组1 - i-1位中比x[i]]小的数之和，记为t[i]，则dp[i]=v[i]*sum{(2*c[i]-(i-1))*x[i]+sum[i-1]-2*t[i]},其中sum[i-1]是前i-1个数的总和。这样共n个状态，每次更新O(logn)，总复杂度：O(nlogn)

树状数组的应用之一：求A数组中前k-1项中比第k项小的元素的和（数组中元素均正）

只要将更新函数由update(x,1)改成update(x,x)即可

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#define maxn 20010
#define lowbit(x) x&(-x)
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;

int n;
struct Cow{
    ll v,x;
    bool operator<(const Cow& rhs)const{
        return v<rhs.v;
    }
}cow[maxn];
ll c1[maxn],c2[maxn],ans,tmp;

void update(ll c[],int i,ll x){
    for(;i<=maxn;i+=lowbit(i)) c[i]+=x;
}

ll sum(ll c[],int i){
    ll s=0;
    for(;i>0;i-=lowbit(i)) s+=c[i];
    return s;
}

int main(){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%lld%lld",&cow[i].v,&cow[i].x);
        }
        sort(cow+1,cow+n+1);
        ans=tmp=0;
        for(int i=1;i<=n;i++){
            ans+=cow[i].v*((2*sum(c1,cow[i].x-1)-(i-1))*cow[i].x+(tmp-2*(sum(c2,cow[i].x-1))));
            tmp+=cow[i].x;
            update(c1,cow[i].x,1);
            update(c2,cow[i].x,cow[i].x);
        }
        printf("%lld\n",ans);
}

bug:int---wa