【训练题43：概率dp】Sample Game | 2021牛客暑期多校训练营4

最新推荐文章于 2024-09-23 11:20:36 发布

溢流眼泪

最新推荐文章于 2024-09-23 11:20:36 发布

阅读量562

点赞数 11

分类专栏：【各类ACM真题】文章标签：概率论动态规划

本文链接：https://blog.csdn.net/weixin_45775438/article/details/119115681

版权

【各类ACM真题】专栏收录该内容

130 篇文章 22 订阅

订阅专栏

题意

Sample Game | 2021牛客暑期多校训练营4
有 $n$ 个数字，分别为 $1\sim n$
选到数字 $i$ 的概率为 $p_i$
每一轮，你都随机选一个数字 $x$ ，然后把它放在自己的集合 $S$ 中
若 $x=\max\{S\}$ ，则你可以继续选数字
否则，你获得分数 $S|^2$
问你获得分数的期望
$1\le n\le 100$

思路

期望倒着推，别想着正着推拉！
期望的设法一般都很单纯(我遇到的题目中)
设 $d p [i]$ 表示开始选到数字 $i$ ，最后的分数？
$\color{red}{No!}$
注意，这里的分数为 $S|^2$
我们有 $e(x^2)\ne e^2(x)$ ，所以我们的设法不能直接这么设！(这个也是很套路的)
设 $d p [i]$ 表示目前选到数字 $i$ ，最后的集合大小
设 $d p 2 [i]$ 表示目前选到数字 $i$ ，最后的集合大小的平方，也就是分数
开始转移！
注意转移也是很套路的。等号后面跟着的就是概率 * (下一个状态 + 贡献)
$dp[i]=\sum_{j=i}^n p_j\times (dp[j]+1)+\sum_{j=1}^{i-1}p_j\times 1\\ dp2[i]=\sum_{j=i}^n p_j\times (dp2[j]+2dp[j]+1)+\sum_{j=1}^{i-1}p_j\times 1^2$
这里又有很多个问题了：
对于 $d p [i]$ 的转移：
（1）下一步有哪些？下一步选的数字为 $j$ ，要么是还能继续选 ( $j\ge i$ ) ，那么期望就是 $p_j(dp[j]+1)$ ，这里 $+ 1$ 是因为你多选了一个数字，集合大小就 $+ 1$ 了；要么是不能选了 ( $j < i$ )，期望就是 $p_j\times1$
（2）等式两边自己包含自己？把右边关于 $d p [i]$ 的系数移到等式左边，然后就能变成 $dp[i]=\frac{Something}{1-p_i}$
对于 $d p 2 [i]$ 的转移：
（1）下一步的贡献怎么算？容易得到 $e((x+1)^2)=e(x^2)+e(2x)+e(1)=e(x^2)+2e(x)+1$
所以中间展开就是这样了，因为我们根据定义，是从 $e(x^2) 推到 e((x+1)^2)$
$e(x^2)\ne e^2(x)$
$e(x^2)\ne e^2(x)$
$e(x^2)\ne e^2(x)$
嗯，好多了…
（2）最后面的是 $p_j\times 1^2$ ，我觉得是这样的，因为下一步无法操作了，贡献就是 $1^2$ 呀
（3）我们发现 $\color{red}{我们算出来的根本就不是集合大小，而是操作次数呀！}$
最后让你算 $e(i^2)$ ，但是相当于我们算的是 $e((i-1)^2)$
嗯？你不会写成答案 = $d p 2 [i] + 1$ 吧 (…)
答案是 $\color{red}{dp2[i]+2dp[i]+1}$

代码

时间复杂度： $O(n^2\log n)$ ，其实可以优化成 $O (n)$ ，无所谓！

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 150;
const int MOD = 998244353;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll qpow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a%MOD;a=a*a%MOD;n>>=1;}return res;}
ll qpow(ll a,ll n,ll p){a%=p;ll res = 1LL;while(n){if(n&1)res=res*a%p;a=a*a%p;n>>=1;}return res;}
ll npow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a;a=a*a;n>>=1;if(res<0||a<0)return 0;}return res;}
ll inv(ll a){/* */return qpow(a,MOD-2);}
ll inv(ll a,ll p){return qpow(a,p-2,p);}

ll dp[MAX],dp2[MAX];
ll p[MAX],w[MAX];
int main()
{
    IOS;
    int n;cin >> n;
    ll sum = 0;
    for(int i = 1;i <= n;++i){
        cin >> w[i];
        sum += w[i];
    }
    for(int i = 1;i <= n;++i){
        p[i] = w[i] * inv(sum) % MOD;
    }
    ll ans = 0;
    for(int i = n;i >= 1;--i){
        ll fz = p[i];
        for(int j = i + 1;j <= n;++j){
            fz = (fz + p[j] * (dp[j] + 1)% MOD) % MOD;
        }
        for(int j = 1;j < i;++j){
            fz = (fz + p[j] * (1) ) % MOD;
        }
        ll fm = (1 - p[i] + MOD) % MOD;
        dp[i] = fz * inv(fm) % MOD;
    }

    for(int i = n;i >= 1;--i){
        ll fz = (p[i] * 2 % MOD * dp[i] % MOD + p[i]) % MOD;
        for(int j = i + 1;j <= n;++j){
            fz = (fz + p[j] * (dp2[j] + 2LL * dp[j] % MOD + 1)% MOD) % MOD;
        }
        for(int j = 1;j < i;++j){
            fz = (fz + p[j] * (1) % MOD) % MOD;
        }
        ll fm = (1 - p[i] + MOD) % MOD;
        dp2[i] = fz * inv(fm) % MOD;
        ans = (ans + (dp2[i] + 2 * dp[i] + 1) % MOD * p[i] % MOD) % MOD;
    }

    cout << ans << endl;
    return 0;
}