POJ 1990 ：MooFest 线段树

$FJ$ 有 $n$ 头牛，排列成一条直线(不会在同一个点)，给出每头牛在直线上的坐标 $x$ 。另外，每头牛还有一个自己的声调 $v$ ，如果两头牛( $i$ 和 $j$ )之间想要沟通的话，它们必须用同个音调 $max(v[i],v[j])$，沟通起来消耗的能量为：$max(v[i],v[j]) * dis(i,j)$ 。问要使所有的牛之间都能沟通(两两之间)，总共需要消耗多少能量。

大概我们可以先把他们按照v从大到小排个序
然后来两个树状数组记录一下他前面有多少头牛，以及牛的横坐标的前缀和
然后……就能算了
然后算完一个删除一个……就行了

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#define N 20000+5
#define M 80000+5
using namespace std;

long long tr[M],sum[M];

struct ask
{
    long long x;
    long long val;
    int pos;
}q[N];

inline int read()
{
    int x=0,f=1;char ch = getchar();
    while(ch < '0' || ch > '9'){if(ch == '-')f=-1;ch = getchar();}
    while(ch >='0' && ch <='9'){x=(x<<1)+(x<<3)+ch-'0';ch = getchar();}
    return x*f;
}

bool cmp1(const ask &a,const ask &b)
{
    return a.x < b.x;
}

bool cmp2(const ask&a,const ask&b)
{
    return a.val > b.val;
}

inline void updata(int k)
{
    tr[k] = tr[k<<1]+tr[k<<1|1];
    sum[k] = sum[k<<1]+sum[k<<1|1];
}

void build(int k,int l,int r)
{
    if(l==r){sum[k]=1,tr[k]=q[l].x;return;}
    int mid = (l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    updata(k);
}

inline void change(int k,int l,int r,int pos)
{
    if(l==r)
    {
        tr[k] = sum[k] = 0;
        return;
    }
    int mid = (l+r)>>1;
    if(pos<=mid)change(k<<1,l,mid,pos);
    else change(k<<1|1,mid+1,r,pos);
    updata(k);
}

int asksum(int k,int l,int r,int x,int y)
{
    if(l==x && r==y)
        return tr[k];
    int mid = (l+r)>>1;
    if(y<=mid)return asksum(k<<1,l,mid,x,y);
    else if(x>mid)return asksum(k<<1|1,mid+1,r,x,y);
    else return asksum(k<<1,l,mid,x,mid)+asksum(k<<1|1,mid+1,r,mid+1,y);
}

int asknum(int k,int l,int r,int x,int y)
{
    if(l==x && r==y)
        return sum[k];
    int mid = (l+r)>>1;
    if(y<=mid)return asknum(k<<1,l,mid,x,y);
    else if(x>mid)return asknum(k<<1|1,mid+1,r,x,y);
    else return asknum(k<<1,l,mid,x,mid)+asknum(k<<1|1,mid+1,r,mid+1,y);
}

int main()
{
    int n = read();
    for(int i=1;i<=n;++i)
        q[i].val=read(),q[i].x=read();
    sort(q+1,q+n+1,cmp1);
    for(int i=1;i<=n;++i)
        q[i].pos = i;
    build(1,1,n);
    sort(q+1,q+n+1,cmp2);
    long long ans = 0;
    for(int i=1;i<=n;++i)
    {
        int num = asknum(1,1,n,1,q[i].pos);
        long long bsum = asksum(1,1,n,1,q[i].pos);
        num--;
        bsum-=q[i].x;
        long long tmp1 = num * q[i].x - bsum;
        long long tmp2 = tr[1]-bsum-q[i].x;
        long long tmp3 = sum[1]-num-1;
        tmp3 *= q[i].x;
        tmp2 -= tmp3;
        tmp1 += tmp2 ;
        ans += q[i].val * tmp1;
        change(1,1,n,q[i].pos);
    }
    cout<<ans<<endl;
}