题意
懒得写了,转换完是求 ( 2 + x + x 2 ) n (2+x+x^2)^n (2+x+x2)n的n-m到n+m次项的系数和(n,m,3e6)
思路
假做法:NTT,只需一次ntt:因为是二次的式子,求点值表达可以直接带数做,点值上做快速幂可以直接得到最终式子的点值表示,最后点值转系数用一次ntt。总复杂度nlogn
正解:枚举原题是每次有六种走法,两种向左,一种不动,三种向右,问最后位移在[-m,m]内有多少种走法。枚举不动的次数,可以转化为求
C
(
n
,
m
)
∗
2
n
∗
3
n
−
m
C(n,m)*2^n*3^{n-m}
C(n,m)∗2n∗3n−m的前缀和,然后递推维护一下这个前缀和即可。
预处理阶乘阶乘逆和23的幂后就是O(n)的了
代码
先咕一下