Boxes （数学期望）

2021牛客暑期多校训练营5

B. Boxes

题意：

​ 求数学期望，n 个箱子，每个箱子里有一个白球或黑球，打开第 i 个箱子的代价为 wi ，可以花费 c 元知道剩余箱子中黑白球的个数，问确定每个箱子中的球，代价的数学期望最小值。

分析：

​ 别问，问就是读假题了 T_T (不用看)

• 首先 C 最多只用花一次，且可以在最开始花，所以有两种策略：
• 直接全部打开，代价：Σwi
• 将 wi 升序排序。先花 C 的代价，剩下的就相当于一个随机 01 序列从前往后开，开到一个 后缀全是同色的为止。
• 总共有 $2^n$ 种情况，遍历枚举后缀同色数 1~n​，后缀同色数为 m 时，情况数为 $2^{n-m}$​ ，对期望的贡献值为$2^{-m}\ast 代价$

Code: (这是错的代码，还莫名AC了)

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 5000005;

double a[N], s[N];

double mn(double a, double b)
{
    if(a<b) return a;
    return b;
}
signed main()
{
	//freopen("1.in","r",stdin);
    //freopen("1.out","w",stdout);
    ios::sync_with_stdio(false);
    int n;
    double c;
    cin>>n>>c;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
    {
        s[i] = s[i-1]+a[i];
    }
    double ans=0, base=0.5;
    int cnt=1;
    for(int i=1;i<n;i++)
    {
        ans += (s[n-i]+mn(c,s[n]-s[n-i]))*base; // 2^(-m) * 代价
        base /= 2;
    }
    ans += mn(c,s[n])*2*base; // 最后全部同色要单独拎出来
    cout<<fixed<<setprecision(12)<<ans;

    return 0;
}

为什么上面的错了？

… … 我开了上帝视角“ ans += (s[n-i]+mn(c,s[n]-s[n-i]))*base ”，mn() 比较之后就知道了要不要先花 C ，并分别对每种情况取最优

从题目来看是两种情况：直接全部打开 and 最开始花C

• 将 wi 升序排序。先花 C 的代价，剩下的就相当于一个随机 01 序列从前往后开，开到一个后缀全是同色的为止。代价：C + Σ wi (1 - 1/$2^{n-i}$​)

Code ：

#include<bits/stdc++.h>
using namespace std;

double w[100005];
int main()
{
	int n;
	double c;
	scanf("%d%lf", &n, &c);
	double sum=0,ans=0;
	for(int i=0;i<n;i++)
		scanf("%lf",&w[i]), sum += w[i];
	sort(w, w+n);
	for(int i=0;i<n;i++)
		ans = ans/2+w[i]; 
	ans = sum-ans+c; // C + Σ wi (1 - 1/2^{n-i}) 
	printf("%.12lf\n", min(ans,sum));
	return 0;
}