CF167B题解

CF167B

这里更容易进入且有翻译

题意

给定初始背包容量 k, 要进行 n 场比赛,每场比赛有 pi% 的概率能够胜利,赢的一场比赛能获得一个奖励——当 ai=1 时获得一个体积为 1 的奖品,或者当 ai>0 时给背包增加 ai 容量,求所有比赛结束后至少赢得 l 场且背包能装下所有奖品的概率。
(1n200,0l,k,ai200,0pi100)

解析

大部分题解都把奖品和增加背包容量放在一起进行转移,其实两者可以分开,我们将能增加背包容量的比赛(以下简称 a 类比赛)设为 ai,其它的比赛(以下简称 b 类比赛)设为 bi。对于背包容量的转移,设 i 为当前进行了几场 a 类比赛(此处转移方程均针对 a 类比赛),j 为已赢下几场比赛,k 为背包容量,有转移方程如下:

dpi,j,k1=dpi1,j,k1×100pi100+dpi1,j1,kai1×pi100

初始化为 dp1,0,k1=1。但此时复杂度为 O(n2ai),约 1.6×109,肯定系T飞了的。

观察到背包容量其实最多只需要到 n,更多的容量没有作用,因此可以将第三维(k 的那一维)的范围从 ai 压缩到 n,大概 200,这样时间复杂度就能过了;另外,对于 dp1 数组的第一维(i 的一维),可以滚动数组优化掉,使空间复杂度从 O(n3) 压到 O(n2),不至于空间超限。

而对于 b 类比赛的转移,由于奖品体积固定为 1,设 i 为第几场 b 类比赛,j 为已赢下 b 类比赛的数量,有转移方程:

dpi,j2=dpi1,j2×100pi100+dpi1,j12×pi100

同样的,dp2 数组的第一维也是可以滚动数组优化掉的。

最后,先对 dp1 数组进行前缀和处理,即可求出答案 ans,设共有 m 场 b 类比赛,答案为

ans=dpm,j2×(dpnm,nm,ai1dpnm,lj1,ai1dpnm,nm,j11+dpnm,lj1,j11)

其中,dpnm,lj1,ai1 在前缀和后变成赢得不超过 lj1 场 a 类比赛的概率,由于 dp1 数组第三维的计算在大概200停止,dpnm,j,ai1 可以用类似于求 dp2 的方法求出。则有

ans=dpm,j2×(1dpnm,lj1,ai1dpnm,nm,j11+dpnm,lj1,j11)

需要注意的是,在计算 dp 数组和答案 ans 的时候,都要注意判定防止数组下标越界(出现负数下标);另外,由于此题对精度的要求仅为 106,dp 时可以用 long long 数组存储而不是 long double,防止丢精度(亲测,被 long double 玩了 QwQ)。

代码

#include <bits/stdc++.h>
#define LL long long
#define double long double
#define pii pair<int, int>

using namespace std;

int ps[205];
vector<int> p;
vector<pii> a;
vector<LL> pon(205, 0.0), powi(205, 0.0); //使用long long防止丢精度
vector<vector<LL> > dp(205, vector<LL>(405, 0)); //dp数组均被滚动优化掉一维

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    p.push_back(0);
    a.push_back({ 0, 0 });

    int n, l, k, c;
    cin >> n >> l >> k;
    for (int i = 1; i <= n; i++)
        cin >> ps[i];
    for (int i = 1; i <= n; i++)
    {
        cin >> c;
        if (c < 0)
            p.push_back(ps[i]);
        else
            a.push_back({ c, ps[i] });
    }

    //计算dp1数组
    dp[0][k] = 1e12;
    for (int i = 1; i < a.size(); i++)
        for (int j = i; j >= 0; j--)
            for (int m = 400; m >= 0; m--)
                dp[j][m] = dp[j][m] * (100 - a[i].second) / 100 + (m >= a[i].first && j ? dp[j - 1][m - a[i].first] * a[i].second / 100 : 0); //三元运算符用于特判防止下标越界,以下同
    for (int i = 1; i <= 400; i++) //前缀和
        dp[0][i] += dp[0][i - 1];
    for (int i = 1; i < a.size(); i++)
    {
        dp[i][0] += dp[i - 1][0];
        for (int j = 1; j <= 400; j++)
            dp[i][j] += dp[i - 1][j] + dp[i][j - 1] - dp[i - 1][j - 1];
    }
    
    //同为dp1数组
    powi[0] = 1e12;
    for (int i = 1; i < a.size(); i++)
        for (int j = i; j >= 0; j--)
            powi[j] = powi[j] * (100 - a[i].second) / 100 + (j ? powi[j - 1] * a[i].second / 100 : 0);
    for (int i = 1; i < a.size(); i++) //前缀和
        powi[i] += powi[i - 1];
        
    //计算dp2数组
    pon[0] = 1e12;
    for (int i = 1; i < p.size(); i++)
        for (int j = i; j >= 0; j--)
            pon[j] = pon[j] * (100 - p[i]) / 100 + (j ? pon[j - 1] * p[i] / 100 : 0);

    //计算答案
    double ans = 0.0;
    for (int i = max(l - (int)a.size() + 1, 0); i < p.size(); i++)
        ans += (double)pon[i] / 1e12 * ((double)((LL)1e12 - (l - i > 0 ? powi[l - i - 1] : 0) - (i ? dp[a.size() - 1][i - 1] : 0) + (l - i > 0 && i ? dp[l - i - 1][i - 1] : 0)) / 1e12);
    printf("%.12LF\n", ans);
}

最后祝各位顺利AC。 >w<

posted @   OEAiHAN  阅读(9)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示