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

【解题思路】：

第i头牛和其他的牛发出的生意就等于=第i头牛的vi*（它之前牛的数目它的坐标-它之前所有牛的坐标和）+vi（它之后所有牛的坐标和-它之后牛的数目*他的坐标），对于牛的数目和牛的坐标，我们都可以用树状数组来维护

【AC代码】：

#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define MOD 1000000009
using namespace std;
inline void read(long long &x){
    char ch=getchar(),c=ch;
	x=0;
    while(ch<'0' || ch>'9'){
    	 c=ch;
		 ch=getchar();
	}
    while(ch>='0' && ch<='9'){
    	x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
    if(c=='-')x=-x;
}

long long wz[20010],yy[20010],n,mn=20000;
long long ans;

struct node{
    long long xi;
    long long vi;
}a[20010];
bool cmp(node x,node y)
{
    return x.vi<y.vi;
}
int lobit(int x) {  return x&(-x);}
void crwz(int x) { for(;x<=mn;x+=lobit(x)) wz[x]++;}
int z(int x)
{
    int sum=0;
    for(;x>=1;x-=lobit(x)) sum+=wz[x];
    return sum;
}
void cryy(int x,int v) { for(;x<=mn;x+=lobit(x)) yy[x]+=v;}
int y(int x)
{
    int sum=0;
    for(;x>=1;x-=lobit(x)) sum+=yy[x];
    return sum;
}
int jdz(int x)
{
    if(x<0) return -x;
    else return x;
}
int main(){
    read(n);
    for(int i=1;i<=n;++i)read(a[i].vi),read(a[i].xi);
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;++i)
    {
            int j=a[i].xi;
            ans+=a[i].vi*(z(j-1)*j-y(j-1)+y(mn)-y(j)-(z(mn)-z(j))*j);
            crwz(a[i].xi);
            cryy(j,a[i].xi);
    }
    printf("%lld",ans);
    return 0;
}