刚刚用 `Python` 做了一下这道题，确实很容易 TLE 和 MLE。 纯 `python3` 做这道题必定TLE，用 `numpy` 来提高速度可过。 如果用 `PyPy3`，速度会加快很多，不过 `PyPy3` 的内存占用比较大，很容易 MLE。可以用 `array` 来开数组，不用 `List`。然后将运算时的中间变量可以复用的复用一下，就可以过。 by irisjiel @ 2022-02-07 23:27:11