bzoj4944 [Noi2017]泳池（dp，概率与期望）

首先把求最大面积恰为K的概率转化成
求最大面积<=K的概率-最大面积<=K-1的概率

我们注意到最底层一定是选了若干段<=K的连续段，于是我们可以枚举连续段的长度来dp。

我们考虑$f[i][j]$表示底边长为i，且每一列都向上延伸至少j的高度且满足面积<=K的概率

那么我们有
$f_{n, m}$ =

$$\begin{cases} 0 & nm > k\\ 1 & n = 0\\ p^nf_{n,m+1}+\sum_{i=0}^{n-1}p^if_{i,m+1}(1-p)f_{n-i-1,m} & \text{default} \end{cases}$$
最后一种转移就是枚举最左的只延伸到m的是哪一列来转移。

然后我们枚举高度来分层dp即可。
复杂度$\sum\limits_{i=1}^k\lfloor \frac{k^2}{i^2}\rfloor=O(k^2)$

可以拿到70分。

剩下的分需要特征多项式，暂时略过了…

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1010
#define mod 998244353
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,f[N],g[N],p,K,bin[N],ans=0;
inline int ksm(int x,int k){
    int res=1;for(;k;k>>=1,x=(ll)x*x%mod) if(k&1) res=(ll)res*x%mod;return res;
}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
int main(){
//  freopen("a3.in","r",stdin);
    n=read();K=read();p=read();p=(ll)p*ksm(read(),mod-2)%mod;bin[0]=1;
    for(int i=1;i<=K;++i) bin[i]=(ll)bin[i-1]*p%mod;g[0]=1;
    for(int h=K;h>=0;--h){//按高度分层dp,f[i]表示底边长为i，且每一列都向上延伸至少h的高度且满足面积<=K的概率 g[i]表示...向上延伸至少h+1的高度...的概率
        memset(f,0,sizeof(f));f[0]=1;
        for(int i=1;i<=(h?K/h:n);++i){
            inc(f[i],(ll)bin[i]*g[i]%mod);
            for(int j=0;j<i;++j) inc(f[i],(ll)g[j]*bin[j]%mod*(1-p+mod)%mod*f[i-j-1]%mod);
        }memcpy(g,f,sizeof(f));
    }ans=g[n];--K;memset(g,0,sizeof(g));g[0]=1;
    for(int h=K;h>=0;--h){
        memset(f,0,sizeof(f));f[0]=1;
        for(int i=1;i<=(h?K/h:n);++i){
            inc(f[i],(ll)bin[i]*g[i]%mod);
            for(int j=0;j<i;++j) inc(f[i],(ll)g[j]*bin[j]%mod*(1-p+mod)%mod*f[i-j-1]%mod);
        }memcpy(g,f,sizeof(f));
    }ans-=g[n];ans%=mod;if(ans<0) ans+=mod;
    printf("%d\n",ans);
    return 0;
}