BZOJ3378:[USACO]MooFest 狂欢节(树状数组)

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

Solution

首先把奶牛按听力排序,那么$max(vi,vj)~(j<i)$一定是$v_i$。问题变成了对于每个$i$,求它和前面奶牛的$dis$和。

开两个树状数组$A$和$B$,$A_i$表示下标为$i$的奶牛的坐标和,$B_i$表示下标为$i$的奶牛的个数,然后算一下就好了。

Code

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #define N (20009)
 5 #define LL long long
 6 using namespace std;
 7 
 8 int n;
 9 pair<int,int>a[N];
10 LL ans,p1,q1,p2,q2;
11 
12 struct BIT
13 {
14     LL c[N];
15     
16     void Update(int x,int k)
17     {
18         for (; x<=20000; x+=(x&-x)) c[x]+=k;
19     }
20     LL Query(int x)
21     {
22         LL ans=0;
23         for (; x; x-=(x&-x)) ans+=c[x];
24         return ans;
25     }
26 }A,B;
27 
28 inline int read()
29 {
30     int x=0,w=1; char c=getchar();
31     while (c<'0' || c>'9') {if (c==-'-') w=-1; c=getchar();}
32     while (c>='0' && c<='9') x=x*10+c-'0', c=getchar();
33     return x*w;
34 }
35 
36 int main()
37 {
38     n=read();
39     for (int i=1; i<=n; ++i) a[i].first=read(), a[i].second=read();
40     sort(a+1,a+n+1);
41     for (int i=1; i<=n; ++i)
42     {
43         int v=a[i].first,x=a[i].second;
44         A.Update(x,x); B.Update(x,1);
45         LL p1=A.Query(x-1),q1=B.Query(x-1);
46         ans+=v*(q1*x-p1);
47         LL p2=A.Query(20000)-A.Query(x),q2=B.Query(20000)-B.Query(x);
48         ans+=v*(p2-q2*x);
49     }
50     printf("%lld\n",ans);
51 }
posted @ 2019-03-12 17:44  Refun  阅读(262)  评论(0编辑  收藏  举报