poj1990两个树状数组

题意:

有一群牛站在一条线上并且他们想要交流;

然而他们都有点聋;

所以他们具备一个能听的声音的最低值v;

并且他们有一个位置x;

题目让你求每两头牛交流之后的和;

这个和的由来是由距离乘以max(v[i], v[j]);

理解:

想了半天,,知道用树状数组;

但是确实不造怎么用;

然后看了大神的理解;

发现要用两个树状数组;

而且还要推一个关系,并且这个关系就是用树状数组维护的;

其意思就是将牛根据v排序,从小到大;

之后可以知道每头牛的max(v[i], v[j])值就是当前牛的v值;

然后计算位置在牛右边的牛的个数,即位置小于当前牛的个数r;

再来计算位置在牛左边的牛的个数,即位置大于当前牛的个数l;

这个就需要用两个树状数组维护数据;

一个维护位置的个数t_bit,一个维护距离c_bit;

个数都好说;

就是要理解距离这一点;

距离的求法可以看这组数据:

2 3 5 7 8

我们求5到前面两个的距离的和,就是5 - 2, 5 - 3.这是O(n);

而用树状数组就是把前面的两个维护了之后可以求到sum = 2 + 3;

之后用 5 * 2 - sum 就求得了距离...

而后面两个是一样的,只不过是调转一下公式罢了, 这样时O(log n);

就这样可以求到当前牛到已经进入树状数组的牛的距离;

然后就可以求得总的和了;

总和的公式就是:

x[i] * (距离) 0 <= i < n;

代码如下:

#include <cstdio>

#include <algorithm>
#include <vector>

using namespace std;

typedef pair<int, int> pii;
typedef long long ll;

#define x first
#define y second

ll sum(vector<ll> &bit, int i)
{
    ll s = 0;
    while (i > 0)
    {
        s += bit[i];
        i -= i & -i;
    }
    return s;
}

void update(vector<ll> &bit, int i, int num)
{
    while (i <= 40000)
    {
        bit[i] += num;
        i += i & -i;
    }
}

int main()
{
    int n;
    scanf("%d", &n);
    vector<pii> v_n(n);
    for (int i = 0; i < n; ++i)
        scanf("%d%d", &v_n[i].x, &v_n[i].y);
    sort(v_n.begin(), v_n.end());

    vector<ll> c_bit(50000, 0), t_bit(50000, 0);  //两个bit
    ll ans = 0;
    for (int i = 0; i < n; ++i)
    {
        ll l = sum(t_bit, v_n[i].y - 1); //左边个数
        ll r = sum(t_bit, 49999) - sum(t_bit, v_n[i].y); // 右边个数
        l = l * v_n[i].y - sum(c_bit, v_n[i].y - 1); // 左边距离
        r = sum(c_bit, 49999) - sum(c_bit, v_n[i].y) - r * v_n[i].y; // 右边距离
        ans += v_n[i].x *(l + r);
        update(t_bit, v_n[i].y, 1);
        update(c_bit, v_n[i].y, v_n[i].y);
    }
    printf("%lld\n", ans);

    return 0;
}