您的位置:首页 > Web前端

POJ1990--POJ 1990 MooFest(树状数组)

2017-08-06 20:42 465 查看
Time Limit: 1000MS
Memory Limit: 30000KTotal Submissions: 8141
Accepted: 3674DescriptionEvery year, Farmer John's N (1 <= N <= 20,000) cows attend "MooFest",a social gathering of cows from around the world. MooFest involves a variety of events including haybale stacking, fence jumping, pin the tail on the farmer, and of course, mooing. When the cows all stand in line for a particular event, they moo so loudly that the roar is practically deafening. After participating in this event year after year, some of the cows have in fact lost a bit of their hearing.
Each cow i has an associated "hearing" threshold v(i) (in the range 1..20,000). If a cow moos to cow i, she must use a volume of at least v(i) times the distance between the two cows in order to be heard by cow i. If two cows i and j wish to converse, they must speak at a volume level equal to the distance between them times max(v(i),v(j)).
Suppose each of the N cows is standing in a straight line (each cow at some unique x coordinate in the range 1..20,000), and every pair of cows is carrying on a conversation using the smallest possible volume.
Compute the sum of all the volumes produced by all N(N-1)/2 pairs of mooing cows. Input* Line 1: A single integer, N
* Lines 2..N+1: Two integers: the volume threshold and x coordinate for a cow. Line 2 represents the first cow; line 3 represents the second cow; and so on. No two cows will stand at the same location. Output* Line 1: A single line with a single integer that is the sum of all the volumes of the conversing cows. Sample Input
4
3 1
2 5
2 6
4 3
Sample Output
57
SourceUSACO 2004 U S Open
题意:给定一个序列,序列两点之间的花费为 两点之间的距离*(两点之间价值大者),求所有N(N-1)/2个点对的花费之和思路:对某个点来说,其等级可以作为乘数的情况是 它与它左右两边比它的等级小的组成点对 设某个点,坐标为 Xc,它的等级为val,其左边比它小的有n1个,右边比它小的有n2个,左边比比它小的坐标为Xi(1<=i<=n1),右边比它小的坐标为 Xj(1<=j<=n2)则经过推导,我们可以得到其对答案的贡献为:
(n1-n2) * Xc * val + val ( segma(Xi) – segma(Xj) )

所以我们需要统计两个类型的值:左右两边比它等级小的点的个数,左右两边比它等级小的点的坐标和。这两个类型的值需要分别求出来,但只要知道了其中一个就可得到另外一个(知道了左边的就可以推导出右边的)我们利用树状数组来统计这些值,具体做法如下:将所有点按照等级排序,则某个点现在的标号即为比它小的所有点(包括左和右)的总个数。我们使用两个树状数组,从左到右一次插入每个节点,一个树状数组统计左边比它等级小的点的个数,另一个统计点的坐标和第一个树状数组好理解,主要困难在于构造第二个树状数组
只需要每次 update(x[i], x[i])即可

这样的话, x[i] 插入之前,getsum(x[i])就是左边比i点等级小的点的坐标和代码:
1 /*
2 * @FileName: D:\代码与算法\2017训练比赛\七月训练四\1013.cpp
3 * @Author: Pic
4 * @Date:   2017-08-04 21:31:01
5 * @Last Modified time: 2017-08-05 20:21:10
6 */
7
8 #include <iostream>
9 #include <algorithm>
10 #include <cstdio>
11 #include <string.h>
12 #include <queue>
13 using namespace std;
14 typedef __int64 ll;
15 const int MAXN=20000+30;
16 //注意,这道题的vol有相同的情况出现。这样的话就不能先按x坐标排序,统计左右比这个点小的个数,然后按照vol排序统计左右比这个点小的点的vol和
17 //而是应该同步地统计这两个量,防止相同的情况
18 struct BIT{
19 	void init()
20 	{
21 		memset(Tree_sum,0,sizeof(Tree_sum));
22 	}
23 	ll Tree_sum[MAXN];//存储树状数组的数组
24 	//int maxn=20000+30;  //树状数组的下标最大值
25 	ll lowbit(ll x)   //lowbit函数, 找到x与与 *最近的一个末位连续0比他多的数* 的距离
26 	{
27 	    return x&(-x);
28 	}
29 	ll getsum(ll x)   //获取0至x区间的和
30 	{
31 	    ll sum=0;
32 	    for(;x>0;x-=lowbit(x)){
33 	        sum+=Tree_sum[x];
34 	    }
35 	    return sum;
36 	}
37 	void update(ll x,ll v)    //更新某点的值
38 	{
39 	    for(;x<=MAXN;x+=lowbit(x)){
40 	        Tree_sum[x]+=v;
41 	    }
42 	}
43 };
44 BIT tr,tr2;
45 struct node
46 {
47 	ll vol,x,id;
48 }a[MAXN];
49 ll n1[MAXN];
50 bool cmp(node a,node b)
51 {
52     return a.vol<b.vol;
53 }
54 bool cmp1(node a,node b)
55 {
56 	return a.x<b.x;
57 }
58 int main(){
59     //freopen("data.in","r",stdin);
60 	ll n;
61 	while(~scanf("%I64d",&n)){
62 		for(int i=0;i<n;i++){
63 			//scanf("%d%d",&a[i].vol,&a[i].x);
64 			//cin>>a[i].vol>>a[i].x;
65 			scanf("%I64d%I64d",&a[i].vol,&a[i].x);
66 			a[i].id=i;
67 		}
68 		tr.init();
69 		tr2.init();
70 		sort(a,a+n,cmp1);
71         sort(a,a+n,cmp);
72         ll res=0,sum=0,sumn=0;
73         for(int i=0;i<n;i++){
74         	sumn=tr.getsum(a[i].x);
75             ll suma=tr2.getsum(a[i].x);
76             res+=((sum-suma-suma)*a[i].vol);
77             res+=((2*sumn-i)*a[i].vol*a[i].x);
78             //cout<<res<<endl;
79             tr2.update(a[i].x,a[i].x);
80             tr.update(a[i].x,1);
81             sum+=a[i].x;
82         }
83         //printf("%lld\n",res);
84         //cout<<res<<endl;
85         printf("%I64d\n",res);
86 	}
87     return 0;
88 }


View Code
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: