[loj2304]泳池
作者：互联网
将等于$k$差分，即小于等于$k$减去小于等于$k-1$，由于两者类似，不妨仅考虑前者
令$f_{i,j}$表示仅考虑$i$列（即$n=i$时），若前$j$行都没有障碍，此时最大面积小于等于$k$的概率
考虑转移，对第$j+1$行是否有障碍分类讨论，并在有障碍时枚举最左边的障碍，即
$$
f_{i,j}=\begin{cases}q^{i}f_{i,j+1}+\sum_{t=1}^{i}(1-q)q^{t-1}f_{t-1,j+1}f_{i-t,j}&(ij\le k)\\0&(ij>k)\end{cases}
$$
初始状态为$f_{0,j}=1$，接下来根据此式子，求出$i,j\ge 1$且$ij\le k$或$j=0$且$i\le k$的$f_{i,j}$，由于这样的状态数是$o(k\log k)$的，总复杂度即为$o(k^{2}\log k)$
接下来，考虑$i>k$且$j=0$时的式子，去掉其中为0的项，即
$$
f_{i,0}=\sum_{t=1}^{k+1}(1-q)q)^{t-1}f_{t-1,1}f_{i-t,0}f_{i,0}=\sum_{t=1}^{k+1}(1-q)q^{t-1}f_{t-1,1}f_{i-t,0}
$$
记$g_{t}=(1-q)q^{t-1}f_{t-1,1}$，即为常数，预处理后即递推式$f_{i,0}=\sum_{t=1}^{k+1}g_{t}f_{i-t,0}$
对于一个多项式$F$，令$G(F)=\sum_{i=0}^{\infty}f_{i,0}[x^{i}]F$（后者即指$F$的$i$次项系数），问题即求$G(x^{n})$
注意到对于函数$G$，有以下两个性质：
1.$G((x^{k+1}-\sum_{i=1}^{k+1}g_{i}x^{k-i+1})x^{t})=f_{t+k+1,0}-\sum_{i=0}^{k}g_{i}f_{t+k+1-i,0}=0$
（即$i=k+t+1$时的递推式，因此显然相等）
2.$G(F_{1})+G(F_{2})=G(F_{1}+F_{2})$，代入即成立
综上，构造$A(x)=x^{k+1}-\sum_{i=1}^{k+1}g_{i}x^{k-i+1}$，设$x^{n}=A(x)B(x)+R(x)$，即有
$$
G(x^n)=G(AB)+G(R)=\sum_{i=0}^{\infty}[x^{i}]B\cdot G(Ax^{i})+G(R)=G(R)
$$
换言之，问题即求$R(x)\equiv x^{n}(mod\ A(x))$，再求$G(R)$即可
若$R_{0}(x)\equiv x^{m}(mod\ A(X))$，即有$R_{0}^{2}(x)\equiv x^{2m}(mod\ A(x))$和$xR_{0}(x)\equiv x^{m+1}(mod\ A(x))$，由此进行迭代即可，复杂度为$o(k^{2}\log k)$（即暴力计算$R_{0}^{2}(x)$）或$o(k\log^{2}k)$（使用ntt计算$R_{0}^{2}(x)$）
综上，总复杂度即$o(k^{2}\log k)$，可以通过
1 #include<bits/stdc++.h>
2 using namespace std;
3 #define N 1005
4 #define mod 998244353
5 int n,k,x,y,p,mi[N],A[N],R[N],RR[N<<1],g[N],f[N][N];
6 int pow(int n,int m){
7 int s=n,ans=1;
8 while (m){
9 if (m&1)ans=1LL*ans*s%mod;
10 s=1LL*s*s%mod;
11 m>>=1;
12 }
13 return ans;
14 }
15 void solve(int n,int k){
16 if (!n){
17 memset(R,0,sizeof(R));
18 R[0]=1;
19 return;
20 }
21 solve((n>>1),k);
22 memset(RR,0,sizeof(RR));
23 for(int i=0;i<=k;i++)
24 for(int j=0;j<=k;j++)RR[i+j]=(RR[i+j]+1LL*R[i]*R[j])%mod;
25 for(int i=2*k;i>k;i--){
26 for(int j=1;j<=k+1;j++)RR[i-j]=(RR[i-j]-1LL*A[k+1-j]*RR[i]%mod+mod)%mod;
27 RR[i]=0;
28 }
29 memcpy(R,RR,sizeof(R));
30 if (n&1){
31 for(int i=k;i>=0;i--)R[i+1]=R[i];
32 R[0]=0;
33 for(int i=1;i<=k+1;i++)R[k+1-i]=(R[k+1-i]-1LL*A[k+1-i]%mod*R[k+1]%mod+mod)%mod;
34 R[k+1]=0;
35 }
36 }
37 int calc(int k){
38 memset(f,0,sizeof(f));
39 for(int i=0;i<=k+1;i++)f[0][i]=1;
40 for(int i=1;i<=k;i++)
41 for(int j=k/i;j>=0;j--){
42 f[i][j]=1LL*mi[i]*f[i][j+1]%mod;
43 for(int t=1;t<=i;t++)f[i][j]=(f[i][j]+1LL*(mod+1-p)*mi[t-1]%mod*f[t-1][j+1]%mod*f[i-t][j])%mod;
44 }
45 for(int i=1;i<=k+1;i++)g[i]=1LL*(mod+1-p)*mi[i-1]%mod*f[i-1][1]%mod;
46 A[k+1]=1;
47 for(int i=1;i<=k+1;i++)A[k-i+1]=mod-g[i];
48 solve(n,k);
49 int ans=0;
50 for(int i=0;i<=k;i++)ans=(ans+1LL*f[i][0]*R[i])%mod;
51 return ans;
52 }
53 int main(){
54 scanf("%d%d%d%d",&n,&k,&x,&y);
55 p=1LL*x*pow(y,mod-2)%mod;
56 mi[0]=1;
57 for(int i=1;i<=k;i++)mi[i]=1LL*mi[i-1]*p%mod;
58 printf("%d",(calc(k)-calc(k-1)+mod)%mod);
59 }
View Code
标签：le,log,int,sum,equiv,loj2304,泳池,mod 来源： https://www.cnblogs.com/PYWBKTDA/p/14734661.html