https://atcoder.jp/contests/abc240/tasks
G - Teleporting Takahashi
从(0,0,0)开始,可以6邻移动1个单位, 不能不动
问 N次后恰好在(x,y,z)的方案数 mod 998244353
范围
n 1e7
x,y,z [-1e7,1e7]
3s
我的思路
感觉有点生成方程 $( x+x^{-1} +y+y^{-1}+z+z^{-1})^n$
然后求$x^Xy^Yz^Z$的系数
如果x选了i 次, 那么x^{-1}选了 i-X 次
$\binom{n}{i}\binom{n-i}{i-X}$
如果y选了j 次, 那么y^{-1}选了 j-Y 次
$\binom{n-2i+X}{j} \binom{n-2i+X-j}{j-Y}$
那么z选了k次, 且z^{-1}选了k-Z次
有 2i-X+2j-Y+2k-Z = n
k = (n+X+Y+Z)/2 -i -j
$\binom{n-2i-2j+X+Y}{k} \binom{n-2i-2j+X+Y-k}{k-Z}$
所以合起来
$= \sum \binom{n}{i}\binom{n-i}{i-X}\binom{n-2i+X}{j}\binom{n-2i+X-j}{j-Y}\binom{n-2i-2j+X+Y}{k}\binom{n-2i-2j+X+Y-k}{k-Z}$
$= \sum \frac{n!}{i!(i-X)!j!(j-Y)!k!(k-Z)!}$
$= \sum \frac{n!}{i!(i-X)!j!(j-Y)!(\frac{n+X+Y+Z}{2}-i-j)!(\frac{n+X+Y-Z}{2}-i-j)!}$
$i \ge max(0,X)$
$j \ge max(0,Y)$
$\frac{n+X+Y+Z}{2} - i - j = k \ge max(0,Z)$
$i+j \le \frac{n+X+Y+Z}{2} - max(0,Z)$
直接算,得n^2会TLE
但如果给定i, 看能不能把j相关的变成一个表达式
$\sum_{j=max(0,Y)}^{\frac{n+X+Y+Z}{2}-max(0,Z)-i} \frac{1}{j!(j-Y)!(\frac{n+X+Y+Z}{2}-i-j)!(\frac{n+X+Y-Z}{2}-i-j)!} $
右侧看起来 是$\frac{1}{j!(j-Y)!}$ 与剩余部分的卷积