Luogu2345 奶牛集会(树状数组)

Posted 2020-11-25 zhwer

题目背景

MooFest, 2004 Open

题目描述

约翰的 (N) 头奶牛每年都会参加“哞哞大会”。哞哞大会是奶牛界的盛事。集会上的活动很多，比如堆干草，跨栅栏，摸牛仔的屁股等等。它们参加活动时会聚在一起，第i 头奶牛的坐标为 (Xi)，没有两头奶牛的坐标是相同的。奶牛们的叫声很大，第i 头和第j 头奶牛交流，会发出 (max[Vi,Vj]×|Xi?Xj|) 的音量，其中 (Vi) 和 (Vj) 分别是第 (i) 头和第 (j) 头奶牛的听力。

假设每对奶牛之间同时都在说话，请计算所有奶牛产生的音量之和是多少。

输入格式

? 第一行：单个整数 (N) ((1 ≤ N ≤ 20000))

? 第二行到第N + 1 行：第i + 1 行有两个整数 (Vi) 和 (Xi) ((1 ≤ Vi ≤ 20000,1 ≤ Xi ≤ 20000))

输出格式

? 单个整数：表示所有奶牛产生的音量之和

输入输出样例

输入 #1
4
3 1
2 5
2 6
4 3
输出 #1
57
————————————————————————————————————
音量的计算公式 (max[Vi,Vj]×|Xi?Xj|) 看起来很友好，

由于奶牛两两之间都会发出声音，很容易想到先将奶牛按 (Vi) 升序排序，之后每遍历到一只奶牛，就计算这只奶牛与此前出现的所有奶牛发出的音量，统计进答案中即可。

这样我们需要求的就是当前奶牛与此前出现的所有奶牛的距离之和，即 (sum_{j=1}^{i-1}dis(i,j))

上面的这个式子是 (O(N^2)) 的，而我们需要在 (O(logn)) 的时间内解决，所以可以使用支持单点修改和区间查询的树状数组优化，

这里使用了两个树状数组，分别存储了奶牛的个数和奶牛的坐标和（即到原点的距离和）

接下里我们把坐标小于当前奶牛和坐标大于当前奶牛的分开处理，具体计算公式请见代码。

最后别忘了把这个值乘上当前奶牛的 (Vi)再 统计进答案，之后还要把当前奶牛的数值加入两个树状数组中。

PS：还要开(long long)。

#include <bits/stdc++.h>
#define FOR(i,s,t) for (ll (i)=(s);(i)<=(t);(i)++)
#define lowbit(x) ((x)&(-x))
#define ll long long
#define MAXN 20007
#define MAXX 20007
using namespace std;
struct cow { ll v,x; }p[MAXN];
ll c1[MAXN],c2[MAXN],n,ans=0;
inline bool cmp(const cow &A,const cow &B) { return A.v<B.v; }
inline void add(ll x) { for (ll i=x;i<MAXX;i+=lowbit(i)) c1[i]++,c2[i]+=x; }
inline ll q1(ll x) {
    ll ret=0;
    for (ll i=x;i>=1;i-=lowbit(i)) ret+=c1[i];
    return ret;
}
inline ll q2(ll x) {
    ll ret=0;
    for (ll i=x;i>=1;i-=lowbit(i)) ret+=c2[i];
    return ret;
}
inline ll read() {
    ll X=0,w=0; char ch=0;
    while (!isdigit(ch)) w|=ch=='-',ch=getchar();
    while (isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
int main() {
    memset(c1,0,sizeof(c1));
    memset(c2,0,sizeof(c2));
    n=read(); FOR(i,1,n) p[i].v=read(),p[i].x=read();
    sort(p+1,p+n+1,cmp);
    FOR(i,1,n) {
        ll amt1=q1(p[i].x-1),dis1=amt1*p[i].x-q2(p[i].x-1);
        ll amt2=q1(MAXX-1)-q1(p[i].x),dis2=q2(MAXX-1)-q2(p[i].x-1)-amt2*p[i].x;
        ans+=(dis1+dis2)*p[i].v,add(p[i].x);
    }
    printf("%lld",ans);
    return 0;
}