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