感性认识一下,发现方案必然是这样的:
也有可能是从下往上一直往右的。
这样我们就有了一种O(N^2)做法:枚举转折点,然后计算最右方块的坐标减去最下方块的坐标。
考虑从上往下枚举转折点。
发现一个方块只有两种放法:往能放的最右放,或者往能放的最左放,也就是说,两种方案坐标的差值为2。
还有一个性质:一个方块放左还是放右,对其下的方块的整体形状没有影响。
可以得到一个式子:从放在一边转到另一边,相当于其下所有方块平移2*当前重量/当前方块及其上的重量之和。
那么就可以直接算了,复杂度O(n)。
代码极短:
#include<bits/stdc++.h>
#define N 300100
using namespace std;
int n;
double a[N],mz[N],xz[N],pla[N];
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%lf",&a[i]);
mz[n-1]=a[n-1];
for(int i=n-2;i>=0;i--){
pla[i]=xz[i+1]/mz[i+1]+1;
mz[i]=mz[i+1]+a[i];
xz[i]=xz[i+1]+pla[i]*a[i];
}
double now=0,ans=pla[0];
for(int i=n-2;i>=1;i--){
now+=2*a[i]/mz[i];
ans=max(ans,pla[i]+pla[0]-now);
}
printf("%lf\n",ans);
return 0;
}