problem

luogu

現(xiàn)有$N(1\le N\le 300)$ 個(gè)盤子，編號(hào)為$1,2,3,\dots ,N$。

第 $i$個(gè)盤中放有 $a_i(1\le a_i\le 3)$個(gè)壽司。

接下來(lái)每次執(zhí)行以下操作，直至吃完所有的壽司。

從第 $1,2,3,\dots ,N$ 個(gè)盤子中任選一個(gè)盤子，吃掉其中的一個(gè)壽司。若沒(méi)有壽司則不吃。

若將所有壽司吃完，請(qǐng)問(wèn)此時(shí)操作次數(shù)的數(shù)學(xué)期望是多少？

solution

最直接地設(shè) $f(a_1,a_2,a_3,...,a_n):$ 第 $i$ 盤還剩 $a_i$ 個(gè)壽司的期望次數(shù)。

那么枚舉隨機(jī)到的盤子，有方程：$f(a_1,a_2,...,a_n)=1+{\sum }_{i=1}^n\frac{1}{n}f(a_1,a_2,...,\max (a_i?1,0),...,a_n)$。

顯然這個(gè)等式不能構(gòu)成轉(zhuǎn)移方程，因?yàn)榇嬖谠卮蜣D(zhuǎn)（第 $i$ 盤壽司為空時(shí)，就變成了自身轉(zhuǎn)移到自身，狀態(tài)不變）的情況。

由于隨機(jī)均勻分布，盤子的位置是不重要的，事實(shí)上我們真正關(guān)注的只有盤子中剩余壽司的數(shù)量。

而 $a_i$ 壽司數(shù)量又只有四種取值 $0/1/2/3$。

不妨重新設(shè) $f(o,i,j,k):$ 當(dāng)前還剩下 $o/i/j/k$ 個(gè)盤子中有 $0/1/2/3$ 個(gè)壽司。

則有轉(zhuǎn)移：
$$f(o,i,j,k)=1+\frac{o}{n}f(o,i,j,k)+\frac{i}{n}f(o+1,i?1,j,k)+\frac{j}{n}f(o,i+1,j?1,k)+\frac{k}{n}f(o,i,j+1,k?1)$$

$$\frac{n?o}{n}f(o,i,j,k)=1+\frac{i}{n}f(o+1,i?1,j,k)+\frac{j}{n}f(o,i+1,j?1,k)+\frac{k}{n}f(o,i,j+1,k?1)$$

$$f(o,i,j,k)=\frac{n}{i+j+k}+\frac{i}{i+j+k}f(o+1,i?1,j,k)+\frac{j}{i+j+k}f(o,i+1,j?1,k)+\frac{k}{i+j+k}f(o,i,j+1,k?1)$$

這樣就不存在狀態(tài)相同的死循環(huán)轉(zhuǎn)移了。

但是現(xiàn)在是 $O(n^4)$ 的，需要進(jìn)一步優(yōu)化。

不難發(fā)現(xiàn)，盤子數(shù)量是固定不變的，即 $o+i+j+k=n$，所以當(dāng)我們知道了其中任意三個(gè)數(shù)，就能推出剩下一個(gè)數(shù)。

設(shè) $f(i,j,k):$ 當(dāng)前還剩下 $i/j/k$ 個(gè)盤子中有 $1/2/3$ 個(gè)壽司。
$$f(i,j,k)=\frac{n}{i+j+k}+\frac{i}{i+j+k}f(i?1,j,k)+\frac{j}{i+j+k}f(i+1,j?1,k)+\frac{k}{i+j+k}f(i,j+1,k?1)$$
最后還要注意循環(huán)枚舉的細(xì)節(jié)：

• $k$ 只用了 $k?1$，當(dāng)在 $(i,j)$ 時(shí)會(huì)問(wèn)到 $j+1$，所以 $k$ 要在 $j$ 循環(huán)的外層。
• 同理 $j$ 會(huì)用到同 $k$ 下的 $j+1$，但此時(shí)是 $i?1$，所以 $j$ 循環(huán)要在 $i$ 的外層。
• 綜上我們確定唯一的循環(huán)順序是 $k,j,i$。

時(shí)間復(fù)雜度： $O(n^3)$。

code

#include <bits/stdc++.h> using namespace std; #define maxn 305 double f[maxn][maxn][maxn]; int a[5]; int n; int main() {scanf( "%d", &n );for( int i = 1, x;i <= n;i ++ ) scanf( "%d", &x ), a[x] ++;for( int k = 0;k <= n;k ++ )for( int j = 0;j <= n;j ++ )for( int i = 0;i <= n;i ++ )if( i or j or k ) {if( i ) f[i][j][k] += f[i - 1][j][k] * i / (i + j + k);if( j ) f[i][j][k] += f[i + 1][j - 1][k] * j / (i + j + k);if( k ) f[i][j][k] += f[i][j + 1][k - 1] * k / (i + j + k);f[i][j][k] += n * 1.0 / (i + j + k);} printf( "%.10f\n", f[a[1]][a[2]][a[3]] );return 0; }

總結(jié)

以上是生活随笔為你收集整理的[AtCoder Educational DP Contest] J - Sushi（期望dp）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。