題目背景

MooFest, 2004 Open

題目描述

約翰的N 頭奶牛每年都會參加“哞哞大會”。哞哞大會是奶牛界的盛事。集會上的活動很

多，比如堆幹草，跨柵欄，摸牛仔的屁股等等。它們參加活動時會聚在一起，第i 頭奶牛的坐標為Xi，沒有兩頭奶牛的坐標是相同的。奶牛們的叫聲很大，第i 頭和第j 頭奶牛交流，會發出max{Vi; Vj}×|Xi − Xj | 的音量，其中Vi 和Vj 分別是第i 頭和第j 頭奶牛的聽力。假設每對奶牛之間同時都在說話，請計算所有奶牛產生的音量之和是多少。

輸入輸出格式

輸入格式：

• 第一行：單個整數N，1 ≤ N ≤ 20000

• 第二行到第N + 1 行：第i + 1 行有兩個整數Vi 和Xi，1 ≤ Vi ≤ 20000; 1 ≤ Xi ≤ 20000

輸出格式：

• 單個整數：表示所有奶牛產生的音量之和

輸入輸出樣例

4
3 1
2 5
2 6
4 3

57

說明

樸素O(N2)

類似於歸並排序的二分O(N logN)

樹狀數組O(N logN)

$$V*+abs(a_1-X)+V*abs(a_2-X)+V*abs(a_3-X)+....
\\=V*(abs(a_1-X)+abs(a_2-X)+abs(a_3-X)+.......)

推公式的題目
設當前奶牛的音量為V,坐標為X,ai表示第i頭奶牛的坐標
假定a1,a2>X,a3<X(方便理解)
我們發現abs不滿足分配率(就是abs(a+b)!=abs(a)+abs(b) )
此時我們分情況討論
設有n個奶牛的坐標比X大，有m個奶牛的坐標比X小,把上面的abs拆開

那麽對於N,M,a1+a2+...,a3+,,,
利用用樹狀數組求逆序對的思想
我們可以用兩個樹狀數組維護

 1 #include<iostream>
 2
 
 #include<cstdio>
 3 #include<cstring>
 4 #include<cmath>
 5 #include<algorithm>
 6 #include<deque>
 7 #include<queue>
 8 #define LL long long 
 9 #define lb(x)    ((x)&(-x))
10 using namespace std;
11 const LL MAXN=40000001;
12 inline LL read()
13 {
14     char c=getchar();LL x=0,f=1;
15     while(c<‘0‘||c>‘9‘)    {if(c==‘-‘)    f=-1;c=getchar();}
16     while(c>=‘0‘&&c<=‘9‘)    x=x*10+c-48,c=getchar();return x*f;
17 }
18 struct node
19 {
20     LL v,x;
21 }cow[MAXN];
22 LL n;
23 int    comp(const node &a,const node &b)
24 {
25     return a.v<b.v;
26 }
27 LL tree_num[MAXN];
28 LL tree_sum[MAXN];
29 LL MAXX;
30 inline void Point_Add(LL pos,LL val,bool how)
31 {
32     while(pos<=MAXX)    
33     {
34         if(how==1)tree_num[pos]+=val;
35         else tree_sum[pos]+=val;
36         pos+=lb(pos);
37     }
38 }
39 inline LL Interval_Ask(LL pos,bool how)
40 {
41     LL ans=0;
42     while(pos)
43     {
44         if(how==1) ans=ans+tree_num[pos];
45         else ans=ans+tree_sum[pos];
46         pos-=lb(pos);
47     }
48     return ans;
49 }
50 int main()
51 {
52     n=read();
53     for(LL i=1;i<=n;i++)    cow[i].v=read(),cow[i].x=read(),MAXX=max(MAXX,cow[i].x);
54     sort(cow+1,cow+n+1,comp);
55     LL ans=0;
56     for(LL i=1;i<=n;i++)
57     {
58         // 1:數量   0：和 
59         ans+=cow[i].v*(  ( -cow[i].x*( Interval_Ask(MAXX,1)-Interval_Ask(cow[i].x,1) ) + (Interval_Ask(MAXX,0)-Interval_Ask(cow[i].x,0)) )+
60                           cow[i].x*Interval_Ask(cow[i].x,1)-(Interval_Ask(cow[i].x,0)) );
61         Point_Add(cow[i].x,1,1);
62         Point_Add(cow[i].x,cow[i].x,0);
63     }
64     printf("%lld",ans);
65     return 0;
66 }

洛谷P2345 奶牛集會