奶牛集会(luogu p2345)

https://www.luogu.org/problem/P2345
题目描述

约翰的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
说明/提示

朴素O(N2)

类似于归并排序的二分O(N logN)

树状数组O(N logN)
为避免v重复而计算错误，需要写两个树状数组
数组a记录个数，数组b纪录数本身，以便统计和

#include <iostream>
#include<algorithm>
#include <stdio.h>
#include <string>
#include <string.h>
#include <map>
#include <math.h>
#include <vector>
#include <set>
#include <queue>
#include <map>
#include <deque>
using namespace std;
typedef  long long ll;
const int N=50010;
ll a[N],b[N];//a为个数，b为和
ll n,m=0;
struct sa
{
    ll v,x;
}p[N];
void update(ll *p,ll x,ll y)
{
    for(;x<=m;x+=(x&-x))
        p[x]+=y;
}
ll sum(ll *p,ll x)
{
    ll ans=0;
    for(;x;x-=(x&-x))
        ans+=p[x];
    return ans;
}
int cmp(sa a,sa b)
{
    return a.v<b.v;
}
int main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++) {
        scanf("%lld %lld", &p[i].v, &p[i].x);
        m=max(m,p[i].x);
    }
    sort(p+1,p+n+1,cmp);//按v从小到大排序
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        ll q=p[i].x;
        ll n1=sum(a,q),n2=sum(a,m)-sum(a,q);//小于q的个数，大于q的个数
        ll a1=sum(b,q),a2=sum(b,m)-sum(b,q);//小于q的和，大于q的和
        ans+=p[i].v*(n1*q-a1+a2-n2*q);//以p[i].v为最大值的结果
        update(a,q,1);
        update(b,q,q);
    }
    printf("%lld\n",ans);
    return 0;
}