洛谷P2345 [USACO04OPEN]MooFest G

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

假设每对奶牛之间同时都在说话，请计算所有奶牛产生的音量之和是多少。
输入
4
3 1
2 5
2 6
4 3
输出
57

结论，如果一个树状数组解决不了题目，那咱们就开俩个~~！（狗头，仅限于这道题）
首先分析一下式子max{Vi; Vj}×|Xi − Xj |
乘号左边取决于两头牛听力较大者，所以我们每次计算的时候，就只计算听力比当前牛牛要小的牛牛所产生的权值部分，我们将牛牛根据听力的值，从小到大进行排序，这样就可以避免漏算和重复算。
然后分析式子的左边。绝对值当然就可以拆成两种情况。Xi<Xj和Xi>Xj，所以一头牛牛对答案的贡献就可以视为如下

V*(num1* X-Xmin+Xmax-num2* X)

num1表示横坐标小于X的牛牛数量
num2表示横坐标大于X的牛牛数量
Xmin表示横坐标小于X的牛牛的距离之和
Xmax表示横坐标大于X的牛牛的距离之和
其中num2又可以用已经出现的牛牛数量减去num1来得到
所以我们开两个树状数组来维护，num和Xmin、Xmax

#include<cstdio>
#include<iostream>
#include<cmath>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf=1000000000;
const ll mod=100003;
struct node {
	ll val,p;
};
struct node a[20004];
ll tree1[20005],tree2[20004];//两个哦
int lowbit(int x){
	return x&(-x);
}
void add1(int now,ll x){
	while(now<=20000){
		tree1[now]+=x;
		now+=lowbit(now);
	}
}

void add2(int now,ll x){
	while(now<=20000){
		tree2[now]+=x;
		now+=lowbit(now);
	}
}
ll ask1(int now){
	ll ans=0;
	while(now>0){
		ans=ans+tree1[now];
		now-=lowbit(now);
	}
	return ans;
}
ll ask2(int now){
	ll ans=0;
	while(now>0){
		ans=ans+tree2[now];
		now-=lowbit(now);
	}
	return ans;
}
bool cmp(struct node x,struct node y){
	return x.val<y.val;
}
int main(){
	#ifdef LOCAL
	freopen(".\\a.in", "r", stdin);
	#endif
	int n;
	memset(tree1,0,sizeof(tree1));
	memset(tree2,0,sizeof(tree2));
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld%lld",&a[i].val,&a[i].p);
	}
	sort(a+1,a+n+1,cmp);//排序
	ll ans=0;
	for(int i=1;i<=n;i++){
		ll tmp1=ask1(a[i].p);//Xmin
		ll tmp2=ask1(20000)-ask1(a[i].p);//Xmax
		ll num=ask2(a[i].p);//num1
		ans=ans+a[i].val*(num*a[i].p-tmp1+tmp2-(i-1-num)*a[i].p);//i-1-num就是上面的num2
		add1(a[i].p,a[i].p);//维护tree1
		add2(a[i].p,1); //维护tree2
	}
	printf("%lld\n",ans);
	return 0;
}