bzoj 3378: [Usaco2004 Open]MooFest 狂欢节（BIT）

3378: [Usaco2004 Open]MooFest 狂欢节

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 88   Solved: 57
[ Submit][ Status][ Discuss]

Description

    每一年，约翰的N(1≤N≤20000)只奶牛参加奶牛狂欢节．这是一个全世界奶牛都参加的大联欢．狂欢节包括很多有趣的活动，比如干草堆叠大赛、跳牛栏大赛，奶牛之间有时还相互扎屁股取乐．当然，她们会排成一列嚎叫，来欢庆她们的节日．奶牛们的叫声实在刺耳，以致于每只奶牛的听力都受到不同程序的损伤．现在告诉你奶牛i的听力为vi(l≤Vi≤20000)，这表示如果奶牛j想说点什么让她听到，必须用高于vi×dis(i，j)的音量．而且，如果奶牛i和j想相互交谈，她们的音量必须不小于max(Vi，Vj)×dis(i，j).其中dis(i，J)表示她们间的距离．

    现在N只奶牛都站在一条直线上了，每只奶牛还有一个坐标xi(l≤xi≤20000)．如果每对奶牛都在交谈，并且使用最小音量，那所有n(n-1)/2对奶牛间谈话的音量之和为多少？

Input

    第1行输入N，之后N行输入Vi和xi．

Output

    输出音量之和．

Sample Input

4

3 1

2 5

2 6

4 3

Sample Output

57

对于牛的听力v[]从大到小排序，然后一个一个算贡献，算完之后就把这个牛剔除

然后对于第i头牛，s1[i]表示在第i头牛前面还有多少头牛，s2[i]表示这些牛的坐标之和

那么在第i头牛前面所有牛对答案的贡献就是s2[i]-s1[i]*v[i]

在第i头牛后面所有牛的答案的贡献算法同理，这个过程可以用树状数组维护

#include<stdio.h>
#include<algorithm>
using namespace std;
#define LL long long
typedef struct Res
{
	int x, y;
	bool operator < (const Res &b) const
	{
		if(y>b.y)
			return 1;
		return 0;
	}
}Res;
Res s[20005];
LL tre[20005];
int m, t2[20005];
LL Query(int x)
{
	LL ans = 0;
	while(x)
	{
		ans += tre[x];
		x -= x&-x;
	}
	return ans;
}
int Query2(int x)
{
	int ans = 0;
	while(x)
	{
		ans += t2[x];
		x -= x&-x;
	}
	return ans;
}
void Update2(int x, int p)
{
	while(x<=m)
	{
		t2[x] += p;
		x += x&-x;
	}
}
void Update(int x, int p)
{
	while(x<=m)
	{
		tre[x] += p;
		x += x&-x;
	}
}
int main(void)
{
	LL ans;
	int n, i;
	m = 20000;
	scanf("%d", &n);
	for(i=1;i<=n;i++)
		scanf("%d%d", &s[i].y, &s[i].x);
	sort(s+1, s+n+1);
	ans = 0;
	for(i=1;i<=n;i++)
	{
		Update(s[i].x, s[i].x);
		Update2(s[i].x, 1);
	}
	for(i=1;i<=n;i++)
	{
		Update(s[i].x, -s[i].x);
		Update2(s[i].x, -1);
		ans += (Query(m)-Query(s[i].x)-(LL)(Query2(m)-Query2(s[i].x))*s[i].x)*s[i].y;
	}
	for(i=1;i<=n;i++)
	{
		Update(s[i].x, m-s[i].x+1);
		Update2(s[i].x, 1);
	}
	for(i=1;i<=n;i++)
	{
		Update(s[i].x, s[i].x-m-1);
		Update2(s[i].x, -1);
		ans += (Query(s[i].x)-(LL)Query2(s[i].x)*(m-s[i].x+1))*s[i].y;
	}
	printf("%lld\n", ans);
	return 0;
}