代码源每日一题 div1 #802 吃蛋糕

802 吃蛋糕

思路

读完题目，以及看到数据范围，大概猜出来，这题是概率dp

观察到$a_i<=3$,所以从这个条件入手。

• $f[i][j][k]$表示，当前有1个蛋糕的盆子有i个，2个蛋糕盆子有j个，3个蛋糕有k个，吃完这些蛋糕的所有方案。
• 设计完状态之后，转移方程就很好实现了，答案就是下面四种情况加起来
  • 吃1个蛋糕的盆子。$i/n\ast f[i-1][j][k]$
  • 2个蛋糕的盆子。$j/n\ast f[i+1][j-1][k]$
  • 3个蛋糕的盆子。$k/n\ast f[i+1][j+1][k-1]$
  • 空盘子。$(n-i-j-k)/n\ast f[i][j][k]$

即$f[i][j][k]=i/n\ast f[i-1][j][k]+j/n\ast f[i+1][j-1][k]+k/n\ast f[i+1][j+1][k-1]+(n-i-j-k)/n\ast f[i][j][k]+1（操作一次）$

注意到两边都有$f[i][j][k]$, 移项，得

$f[i][j][k]=(i\ast f[i-1][j][k]+j\ast f[i+1][j-1][k]+k\ast f[i+1][j+1][k-1])/(i+j+k)+n/(i+j+k)$

code

#include<bits/stdc++.h>
using namespace std;
const int N = 305, M = 2 * N;  
typedef long long LL;
typedef pair<int,int> PII;
using tp = tuple<int,int,int>;
bool multi = false;

int n;
double f[N][N][N];
int cnt[4];

double dp(int a, int b, int c) {
    if(f[a][b][c] >= 0) return f[a][b][c];
    int s = a + b + c;
    if(!s) return 0;
    f[a][b][c] = (double)n / s;
    if(a) f[a][b][c] += a * 1.0 / s * dp(a - 1, b, c);
    if(b) f[a][b][c] += b * 1.0 / s * dp(a + 1, b - 1, c);
    if(c) f[a][b][c] += c * 1.0 / s * dp(a, b + 1, c - 1);
    return f[a][b][c];
}
void solve() {
    memset(f, -1, sizeof f);
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        int x;
        scanf("%d", &x);
        cnt[x]++;
    }

    printf("%.9lf\n", dp(cnt[1], cnt[2], cnt[3]));
}   

int main()
{
#ifdef ONLINE_JUDGE
#else 
    freopen("I.txt", "r", stdin);
#endif
    int T = 1;
    if(multi) cin >> T;
    while(T--) solve();
    return 0;
}