POJ

题目链接

题意：

给定 $N$ 头牛的位置$(x)$和听力阈值$(v)$，每两头牛之间交流需要的声音是 $max( v1, v2) * | x1 - x2 |$ ，现在要求所有牛之间交流的声音总值。

思路：

对于每两头牛之间，总是取最大的$V$ , 我们就可以按照$V$的大小排序；

那么我们就可以用当前牛的 $V *$ 已经存在的牛到当前牛的距离总和 ，并更新答案。

这里要用树状数组记录一下  下标之和 $and$ 个数 ，于是我们就两个树状数组。

/*
* @Author: windystreet
* @Date:   2018-08-07 15:13:40
* @Last Modified by:   windystreet
* @Last Modified time: 2018-08-07 15:52:16
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>

using namespace std;

#define X first
#define Y second
#define eps  1e-5
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!
");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 2e4+7;
const int INF  = 1<<30;
const int mod  = 1e9+7;

struct node
{
    int x,v;
}s[maxn];

LL tree[maxn][3];     // 0 下标和
int n;                // 1  个数

bool cmp(node a,node b){
    return a.v<b.v||(a.v==b.v&&a.x<b.x);
}


void add(int x,int v,int flag){
    for(int i=x;i<maxn;i+= lowbit(i)){
        tree[i][flag] += v;
    }
    return ;
}
LL query(int x,int flag){
    LL res = 0;
    for(int i=x;i;i-=lowbit(i)){
        res += tree[i][flag];
    }
    return res;
}


void solve(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&s[i].v,&s[i].x);
    }
    sort(s+1,s+1+n,cmp);
    LL ans = 0,a1,a2,b1,b2;
    for(int i=1;i<=n;i++){
        a1 = query(s[i].x,0);         // s[i].x之前的牛的下标和
        a2 = query(maxn-1,0) - a1;    // 所有的牛的下标和
        b1 = query(s[i].x,1);         // s[i].x之前的牛的数量
        b2 = query(maxn-1,1) - b1;    // 所有牛的数量
        ans += (b1*s[i].x - a1 + a2 - s[i].x * b2) * s[i].v;  
        add(s[i].x,s[i].x,0);         // b1 * s[i].x - a1 表示s[i].x 之前的牛到s[i].x 的距离之和
        add(s[i].x,1,1);              // a2 - s[i].x * b2 表示s[i].x 之后的牛到s[i].x 的距离之和

    }
    printf("%lld
",ans);

    return;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    //scanf("%d",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}