題目鏈接

http://uoj.ac/contest/51/problem/514

題解

神仙們都好強啊。

本題有好多做法，但是第一步都是一樣的：
題目中的“每次選一個沒有達到 \(a\) 的進行裝填”其實沒有用，可以等價成每次隨機選任何一個位置 \(+1\)，然后求 \(\ge a\) 的個數的期望。
然后考慮計算 \(1\) 號位置最后達到 \(a\) 了的概率。

不容斥做法

考慮操作序列，對每個位置 \(+1\) 視為一種不同的操作。我們將一種非 \(1\) 操作的次數達到 \(b\) 或者 \(1\) 操作達到 \(a\) 稱為“終止”。一種操作終止之后，我們忽略這種操作，不再加入進操作序列中。于是操作序列的總長是 \((n-1)b+a\).
考慮計算最后一次為 \(1\) 操作的概率。那么也就是說剩下 \((n-1)\) 種操作要在之前全部終止。設終止的時間從小到大排序為 \(t_1,t_2,...,t_{n-1}\)，那么容易得到概率為

\[\frac{\prod^{n-1}_{i=1}{t_i-1-(i-1)b\choose b-1}}{\prod^{n-1}_{i=1}(n-i+1)^{t_i-t_{i-1}}} \]

直接對這個東西做一個 DP，設 \(f[i][j]\) 表示目前操作序列放了 \(i\) 個元素，已經有 \(j\) 種操作終止了。
時間復雜度 \(O(n^3)\).

容斥做法

容斥。考慮枚舉 \(1\) 號位置達到 \(a\) 時沒有達到 \(b\) 的位置個數 \(i\). 那么別的位置照樣可以不考慮。
假設欽定了 \(k\) 個位置，連帶著 \(1\) 實際被操作的次數分別記為 \(x_i\)，其中 \(1\) 的次數記作 \(x_0\)，則 \(\forall 1\le i\le k,0\le x_i\le b-1\). 考慮計算當 \(x_0=a-1\) 的時候，出現(xiàn)每個局面的概率之和，再乘以下一步操作 \(1\) 的概率 \(\frac{1}{k}\). 則對于一組合法的 \(x_0=a-1,x_1,...,x_k\)，概率為

\[\frac{(\sum^k_{i=1}b_i)!}{\prod b_i!k^{\sum^k_{i=1}b_i}} \]

對所有欽定方案的所有 \(b\) 序列求和，直接 DP 可以做到 \(O(n^4)\)，NTT 優(yōu)化可以做到 \(O(n^3\log n)\).

有一個神奇的優(yōu)化：發(fā)現(xiàn)我們就是要求一個生成函數 \(F(x)\sum^{b-1}_{i=0}\frac{x^i}{i!}\) 的 \(0,1,...,n\) 次冪的每一項系數。而這種長得很像 \(e^x\) 的函數，求導往往有奇效：

\[(F(x)^i)'=iF(x)^{i-1}F'(x)=iF(x)^i-iF(x)^{i-1}\cdot \frac{x^{b-1}}{(b-1)!} \]

這個式子可以一項一項地推出來系數，時間復雜度 \(O(n^3)\).

然后還有一種神奇的 DP，大概是從前往后考慮操作序列，減掉當前這個數出現(xiàn)次數 \(\ge b\) 的情況。時間復雜度 \(O(n^3)\).

代碼

不容斥做法

#include<bits/stdc++.h> #define llong long long #define mkpr make_pair #define x first #define y second #define iter iterator #define riter reversed_iterator #define y1 Lorem_ipsum_dolor using namespace std;inline int read() {int x = 0,f = 1; char ch = getchar();for(;!isdigit(ch);ch=getchar()) {if(ch=='-') f = -1;}for(; isdigit(ch);ch=getchar()) {x = x*10+ch-48;}return x*f; }const int mxN = 250; const int P = 998244353;void updsum(llong &x,llong y) {x+=y-P,x+=(x>>31)&P;}llong quickpow(llong x,llong y) {if(y==0) {return 1ll;}llong ret = quickpow(x,y>>1); ret = ret*ret%P; if(y&1ll) {ret = ret*x%P;}return ret; } llong fact[mxN*mxN+3],facti[mxN*mxN+3],inv[mxN*mxN+3]; void initfact(int n) {fact[0] = 1ll; for(int i=1; i<=n; i++) fact[i] = fact[i-1]*i%P;facti[n] = quickpow(fact[n],P-2); for(int i=n-1; i>=0; i--) facti[i] = facti[i+1]*(i+1ll)%P;for(int i=1; i<=n; i++) inv[i] = facti[i]*fact[i-1]%P; } llong comb(llong x,llong y) {return x<0||y<0||x<y?0ll:fact[x]*facti[y]%P*facti[x-y]%P;}llong f[mxN*mxN+3][mxN+3]; int n,a,b; llong ans;int main() {initfact(mxN*mxN);n = read(),a = read(),b = read();f[0][0] = 1ll;for(int i=0; i<=(n-1)*b+a-1; i++){for(int j=0; j<=n-1&&j*b<=i; j++){llong x = f[i][j]; if(!x) continue; // printf("f[%d][%d]=%lld\n",i,j,x);x = x*inv[n-j]%P;updsum(f[i+1][j],x);updsum(f[i+1][j+1],x*comb(i-j*b,b-1)%P);}}ans = f[(n-1)*b+a-1][n-1];ans = (1ll-ans*fact[n-1]%P+P)*n%P;printf("%lld\n",ans);return 0; }

總結

以上是生活随笔為你收集整理的UOJ #514 [UR #19]通用测评号 (容斥原理、DP)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。