poj 1990 MooFest(树状数组）

这个题目和寒假牛客第三场的G非常类似。

传送门

题目大意：每头牛都有一个听力值，还有它所在位置。两头牛交流所消耗的能量为两头牛距离差值*两头牛中听力值最大的一个。我们要求任意两头牛交流所需要的的能量和。共n*(n-1）/2。

可以用树状数组维护当前牛之前的牛的个数，和当前牛之前的牛的坐标和。当然首先需要根据听力值排序，这样就保证了能量和最小。

然后逐个得添加到树状数组中然后查询就可以了。查询的时候要注意查询当前牛之前的牛和当前牛之后的牛。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <stack>
#include <map>
#include <sstream>
#include <cstring>
#include <set>
#define IO                       \
    ios::sync_with_stdio(false); \
    // cin.tie(0);                  \
    // cout.tie(0);
using namespace std;
typedef long long LL;
const int maxn = 3e4 + 10;
const int maxm = 1e5 + 10;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
struct Node
{
    int v, x;
} a[maxn];
LL c[2][maxn]; // 0 维护牛的个数 1 维护 坐标和
int n;
int lowbit(int x)
{
    return x & (-x);
}
bool cmp(Node a, Node b)
{
    return a.v < b.v;
}
void add(int k, int p, int val)
{
    int i = p;
    while (i <= maxn)
    {
        c[k][i] += val;
        i += lowbit(i);
    }
}
LL ask(int k, int p)
{
    int i = p;
    LL ans = 0;
    while (i >= 1)
    {
        ans += c[k][i];
        i -= lowbit(i);
    }
    return ans;
}
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
#endif
    IO;
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i].v >> a[i].x;
    sort(a + 1, a + n + 1, cmp);
    LL ans = 0;
    for (int i = 1; i <= n; i++)
    {
        LL sum1 = ask(0, a[i].x);        // 查询当前在你之前的牛的个数
        LL sum2 = ask(1, a[i].x);        // 查询当前牛之前的坐标和
        LL sum3 = ask(1, int(2e4 + 10)); // 当前全部的牛的坐标和

        ans += (a[i].x * sum1 - sum2 + sum3 - sum2 - (i - 1 - sum1) * a[i].x) * a[i].v;
        add(0, a[i].x, 1);
        add(1, a[i].x, a[i].x);
    }
    cout << ans;
    return 0;
}