Educational DP Contest——J 期望dp
题目
https://atcoder.jp/contests/dp/tasks/dp_j
思路
观察到每个盘子的寿司数量最多为3,那么我们定义,表示1,2,3个寿司的盘子数量还有b,c,d的期望,0个的用a表示
根据当前这一次要么选0个寿司的盘子,要么选1个寿司的盘子,要么选2个寿司的盘子,要么选3个寿司的盘子。那么可以得到转移方程:
化简后得到:
枚举的时候b,c,d对应i,j,k。我们需要注意枚举顺序
- 用到了 ,那么 一定得再 的外层,不然你怎么用到
- 用到了 ,那么 一定得在 的外层
- 所以枚举顺序为:
AC代码
点击查看代码
#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x; i<=y; ++i)
#define per(i,x,y) for(int i=x; i>=y; --i)
#define pushk push_back
#define popk pop_back
#define mem(a,b) memset(a,b,sizeof a)
#define ll long long
#define emk emplace_back
using namespace std;
const int N = 300+9;
double dp[N][N][N];
int cnt[4];
int a[N];
int main() {
int n;
cin>>n;
rep(i,1,n) cin>>a[i],cnt[a[i]]++;
dp[0][0][0]=0;
rep(k,0,n){
rep(j,0,n){
rep(i,0,n){
int t = i+j+k;
if(i) dp[i][j][k]+=i*1.0/t*dp[i-1][j][k];
//(i,j)用到了(i+1,j-1),所以j要在i前面
if(j) dp[i][j][k]+=j*1.0/t*dp[i+1][j-1][k];
// (j,k)用到了(j+1,k-1),所以k要在j的外层
if(k) dp[i][j][k]+=k*1.0/t*dp[i][j+1][k-1];
if(t) dp[i][j][k]+=n*1.0/t;
//cout<<dp[i][j][k]<<endl;
}
}
}
printf("%.10f\n",dp[cnt[1]][cnt[2]][cnt[3]]);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话