【补题】运动的球球

思路：

对于一个球来说，如果有一个球x比它小，但速度比他快，那么它们一定会相遇，但是如果速度比它慢，那么一定不会相遇，它们之间的最短距离就是没运动的时候的距离，因此考虑将这些球按x轴的顺序排序。排完序后，对于每个球，我们只要看它左边有多少个比它速度慢的球，还要知道这些球在x轴上的距离是多少，题解给的方式是二维的树状数组，这里叫它a[i][j]，i表示的是离散化后的速度，j表示两个信息， 当j为0时，表示此时比i速度慢的小球个数， 为1时，表示此时比i速度慢的x轴坐标的总和。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+100;

struct boll{
    int x, v;
    bool operator < (const boll &a) const
    {
        if(x==a.x) return v<a.v;
        return x<a.x;
    }
}b[N];

int n, m;
ll a[N][2];
int v[N];

int lowbit(int x)
{
    return x & (-x);
}

void add(int pos, int val)
{
    while(pos<=m)
    {
        a[pos][0]++;
        a[pos][1]+=(ll)val;
        pos+=lowbit(pos);
    }
}

ll get(int pos, int x)
{
    ll res=0;
    while(pos)
    {
        res+=a[pos][x];
        pos-=lowbit(pos);
    }
    return res;
}

int main(){
    scanf("%d", &n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d", &b[i].x);
    }
    for(int i=1; i<=n; i++)
    {
        scanf("%d", &b[i].v);
        v[i]=b[i].v;
    }
    sort(b+1, b+n+1);
    sort(v+1, v+n+1);
    m=n;
    while(m!=lowbit(m)) m+=lowbit(m);
    ll ans=0;
    for(int i=1; i<=n; i++)
    /*
    因为是按照x的坐标从小到大来看的，所以树状数组里存的小球信息都是在当前的小球之前
    */
    {
        int pos=upper_bound(v+1, v+n+1, b[i].v)-(v+1);
        ans+=(ll)b[i].x*get(pos, 0)-get(pos, 1);
        //get(pos, 0)表示比当前小球速度慢并且在当前小球前面的小球个数
        //get(pos, 1)表示比当前小球速度慢并且在当前小球前面的小球x轴坐标之和
        add(pos, b[i].x);//把当前小球的信息记录在树状数组里
    }
    
    printf("%lld\n", ans);
    
    return 0;
}