DTOJ 4872. 真的

题意

你正在一个数轴上随机游走。一开始，你在位置 $0$。每一时刻，你有 $ \frac{1}{2}$ 的概率往左走，还有 $\frac{1}{2}$的概率往右走。定义一个方向系数$t$，当你往左走时 $t$ 为 $-1$，反之往右走 $t$ 为 $+1$。同时，我们给定了一个长度为 $N$ 的序列 $A_1,A_2,\dots ,A_N$ 。假如当前在位置 $x$ ，那么在选定方向之后，下一步有 $\frac{A_k}{{\sum }_{i=1}^N{A}_i}$ 的概率在 $x+t\times k$。

假如走完之后，你落在了某一个位置 $y<0$，那么你会直接到达 $-y$ 的位置；假如你落在了某一个位置 $y>N$，那么你会直接到达 $2\times N-y$ 的位置。

现在你按照上述规则，随机游走了$K$步。请你求出，你最终到达每个点的概率在模 $998244353$ 意义下的结果。显然，最终到达的一定是 $0$ 到 $N$ 之间的某一个整点。

Subtask 1 (3pts)：$K=0$。

Subtask 2 (12pts)：$N\le 100,K\le 5000$

Subtask 3 (23pts)：$N\times K\le 5\times 10^5$

Subtask 4 (11pts)： $ \forall 2 \le i \le N,A_i=0$

Subtask 5 (27pts)： $N\le 10^3$

Subtask 6 (24pts)：无特殊限制。

对于全部数据： $1\le N\le 10^5,0\le K\le 10^{18},1\le {\sum }_{i=1}^N{A}_i\le 10^8$

题解

考虑游走的过程，发现是一个碰壁就回退的循环，考虑构造循环卷积。如果循环节是n显然不太行，发现从$0$到$n$再回到$0$可看作一个循环节，于是构造一个长度为$2n$的多项式，$0$到$n$项依次是$0,...,n$代表从$0$走到$n$，后$n-1$项依次是$n-1,...,1$代表从$n$走到$0$。由于两个方向的概率都是$1/2$，这样向原来方向走相当于右移，向相反方向走相当于左移，如果越界的话则在$\%2n$的意义下是等价的，故直接循环卷积快速幂即可。