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;
}