pku1990 树状数组。。两个树状数组

 

 话说这题正好是我出生年啊。。。

 

 

题意：给一个20000个元素的数组。。。对每个元素有两个性质，深度v和位置x，对任意两个元素i，j有一个值=max(vi, vj)*|xi-xj| 求所有i j之间的值的和。。。

 

 

 

分析：因为对每两个元素都是取v值最大的一个来算，我们索性先按照v从小到大排序，然后对j位置而言，与前面的任意i位置而言都乘以vj。。。这时候我们就要算出前面比j大的元素的x的和，然后减去大于j的x的个数*xj。。。小于的同理。。。所以用两个树状数组，一个维护个数，一个维护和。。。

 

 

代码：

#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;

const int N=20010;
int n, m;
__int64 num[N], sum[N];
__int64 ans;
struct node 
{
	__int64 v, i;
} a[N];

int cmp(node &a, node &b)
{
	return a.v<b.v;
}

int query(int i)
{
	int tmp = 0;
	for(; i>0; i-=i&(-i))
		tmp += sum[i];
	return tmp;
}
void update(int i, int v)
{
	for(; i<N; i+=i&(-i))
		sum[i] += v;
}
int query1(int i)
{
	int tmp = 0;
	for(; i>0; i-=i&(-i))
		tmp += num[i];
	return tmp;
}
void update1(int i, int v)
{
	for(; i<N; i+=i&(-i))
		num[i] += v;
}

int main()
{
	int i, j;
	while(scanf("%d", &n)!=EOF)
	{
		for(i=0; i<n; i++)
		{
			scanf("%I64d%I64d", &a[i].v, &a[i].i);
		}
		sort(a, a+n, cmp);
		ans = 0;
		for(i=0; i<n; i++)
		{
			ans += ( query(N-1)-query(a[i].i)-a[i].i*(query1(N-1)-query1(a[i].i) ) + a[i].i*query1(a[i].i-1)-query(a[i].i-1) )*a[i].v;
			update(a[i].i, a[i].i);
			update1(a[i].i, 1);
		}
		printf("%I64d\n", ans);
	}
	return 0;
}