首页 > 代码库 > HDU 4923 (杭电多校 #6 1003题)Room and Moor(公式+栈)
HDU 4923 (杭电多校 #6 1003题)Room and Moor(公式+栈)
题目地址:HDU 4923
比赛的时候脑残了。。思路完全想出来了。。只不过想出了个根本不可能存在的极端数据,然后一看输入数据是100组,就把自己给否决了。。。sad。。当时就应该大胆试一试的。。。
这个题首先可以把最前面的0和最后面的1去掉,因为这两块总可以用0和1抵消掉。然后中间就分成了10相间的情况,然后根据10相间,可以分成若干组,每一组都是由几个1和几个0组成的。比如说1101101110,就可以分成110,110,1110这样的三组。
然后这时候可以可以对每一组内只取一个数来使得这组的和最小。那应该怎么找这个数呢。假设这组有a个1,b个0,取的数为X,就可以列出式子(a+b)x^2-2*a*x+a,对他求导得知当x为a/(a+b)的时候最小。所以这一段应该取a/(a+b).
然后剩下的只有一个任务了,怎么让某些组合在一块保证所取得这个数是递增的。为了方便,可以弄一个栈。然后当栈为空的时候就放进去。遍历这些组。当该组的x比栈顶的x小,这组就跟前面的那组合并,并把栈顶的弹出。合并完后的值继续与栈顶的比较,直到栈为空或者栈顶的x比该组的x小,再放入栈。如果一上来就比栈顶的大,那就直接放入栈。
最后,由于每组取的值都是a/(a+b),可以代入列出的式子。求得最终的值是(a*b)/(a+b)。可以直接利用这个值来计算。
注意中间过程可能会爆int,要用int64。代码如下:
#include <iostream> #include <cstdio> #include <cstring> #include <math.h> #include <stack> using namespace std; __int64 a[110000], b[110000]; struct node { __int64 l, r, sum, a; double z; }q[110000]; int main() { __int64 t, n, i, j, pos1, pos2, cnt, flag; double ans; scanf("%I64d",&t); while(t--) { stack<node>Q; scanf("%I64d",&n); for(i=0;i<n;i++) { scanf("%I64d",&a[i]); } for(i=0;i<n;i++) { if(a[i]) { pos1=i; break; } } for(i=n-1;i>=0;i--) { if(a[i]==0) { pos2=i+1; break; } } flag=0; cnt=0; int b=0, c=0; for(i=pos1;i!=pos2;) { b=0; c=0; q[cnt].l=i; while(a[i]==1) { i++; b++; } while(a[i]==0&&i<n) { i++; c++; } q[cnt].a=b; q[cnt].r=i-1; q[cnt].sum=q[cnt].r-q[cnt].l+1; q[cnt++].z=b*1.0/(b+c); } for(i=0;i<cnt;i++) { if(Q.empty()) { Q.push(q[i]); continue ; } while(!Q.empty()&&q[i].z<=Q.top().z) { node f1=Q.top(); q[i].l-=f1.sum; q[i].sum+=f1.sum; q[i].a+=f1.a; q[i].z=q[i].a*1.0/q[i].sum; Q.pop(); } Q.push(q[i]); } ans=0; while(!Q.empty()) { node f1=Q.top(); ans+=f1.a*(f1.sum-f1.a)*1.0/f1.sum; Q.pop(); } printf("%.6lf\n",ans); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。