luogu P2345 奶牛集会 |排序+树状数组

Posted 2020-11-23 naruto-mzx

题目描述

约翰的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

输出格式

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

---

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
const int N=2e4+1;
struct E{
    int v,x;
}e[N];
int n;
bool cmp(E t1,E t2){
    return t1.v<t2.v;
}
ll c1[N],c2[N];
void add1(int x,int y)
{
    for(;x<=N;x+=x&(-x))
    c1[x]+=y;
}
ll num1(int x)
{
    int ans=0;
    for(;x;x-=x&(-x))ans+=c1[x];
    return ans;
}
void add2(int x,int y)
{
    for(;x<=N;x+=x&(-x))
    c2[x]+=y;
}
ll num2(int x)
{
    int ans=0;
    for(;x;x-=x&(-x))ans+=c2[x];
    return ans;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    scanf("%d%d",&e[i].v,&e[i].x);
    sort(e+1,e+1+n,cmp);
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        ll p1=num1(e[i].x),p2=num2(e[i].x);
        ans+=e[i].v*abs(e[i].x*p1-p2);
        p1=num1(N)-p1;p2=num2(N)-p2;
        ans+=e[i].v*abs(e[i].x*p1-p2);
        add1(e[i].x,1);
        add2(e[i].x,e[i].x);
    }
    cout<<ans<<endl;
}