题目大意
给定 p , q , α , β p,q,\alpha,\beta p,q,α,β ,以及 f i , 0 = a i , f 0 , i = b i f_{i,0}=a_i,f_{0,i}=b_i fi,0=ai,f0,i=bi ,设 γ = p α + q β \gamma=p\alpha+q\beta γ=pα+qβ 。
f i , j = p f i − 1 , j + q f i , j − 1 + γ f_{i,j}=pf_{i-1,j}+qf_{i,j-1}+\gamma fi,j=pfi−1,j+qfi,j−1+γ, 已知 h h h,求
∑ i = 0 n ∑ j = 0 n f i , j × h i ( n + 1 ) + j \sum_{i=0}^n\sum_{j=0}^nf_{i,j}\times h^{i(n+1)+j} i=0∑nj=0∑nfi,j×hi(n+1)+j
模数为998244353
题解
考虑将 f f f 分开三部分处理。
求 γ \gamma γ 的贡献
容易发现一个点 ( i , j ) (i,j) (i,j) 给 ( x , y ) (x,y) (x,y) 的贡献其实就是 p p p 的向下走的步数次方乘 q q q 的向右走的步数次方再乘上路径数。即
∑ x = 1 n ∑ y = 1 n ∑ i = 0 x − 1 ∑ j = 0 y − 1 γ ( i + j i ) p i q j h x ( n + 1 ) + y \sum_{x=1}^n\sum_{y=1}^n\sum_{i=0}^{x-1}\sum_{j=0}^{y-1}\gamma\begin{pmatrix} i+j\\i \end{pmatrix}p^iq^jh^{x(n+1)+y} x=1∑ny=1∑ni=0∑x−