首页 > 代码库 > CodeForces-28C-Bath Queue-概率DP[ ICPC大连热身D]
CodeForces-28C-Bath Queue-概率DP[ ICPC大连热身D]
题目地址 http://codeforces.com/problemset/problem/28/C
代码+注释
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> using namespace std; //前i个澡堂,一共j个人,最长队伍为k的概率 //dp[i][j][k] double dp[55][55][100]; int a[55]; double C[55][55]; void init() { C[0][0] = 1; for ( int i = 1; i <= 50; i++ ) { C[i][0] = 1; for (int j = 1; j <= i; j++) C[i][j] = C[i-1][j-1] + C[i-1][j]; } } int main() { int m,n; scanf("%d%d",&n,&m); init();//求组合数 memset(dp,0,sizeof(dp)); for ( int i = 1; i <= m ; i++ ) scanf("%d",&a[i]); dp[0][0][0] = 1; // dp初始化 for (int i = 1 ; i <= m; i++) { for (int j = 0; j <= n ; j++) { for ( int k = 0; k <= n; k++) { //枚举[i][j][k] //计算dp[i][j][k] int top = min(k*a[i],j); //只有当前澡堂最长队伍小于等于k时。才能转移 for (int c = 0 ; c <= top ; c++) //i澡堂中人数 { if (k*a[i]-a[i]+1 > c) // 如果当前 澡堂最大值不是k 只能从dp[i-1][j-c][k]转移 dp[i][j][k] += dp[i-1][j-c][k] * pow(i-1,j-c) / pow(i,j) * C[j][c]; else // 如果当前为k,可以从 dp[i-1][j-c][<=K]转移 for (int tt = 0; tt<=k ;tt++) dp[i][j][k] += dp[i-1][j-c][tt] * pow(i-1,j-c) / pow(i,j)* C[j][c]; } } } } double ans = 0; for (int i = 1 ; i <= n; i++) { ans += dp[m][n][i] * i; } printf("%.10f\n",ans); return 0; }
CodeForces-28C-Bath Queue-概率DP[ ICPC大连热身D]
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。