如果你加够了足够的取余但仍然 WA，可能你的答案输出的是负数，要 ans=(ans%mod+mod)%mod。

原因是把 $(a_i+x)^2$ 拆成 $a_i^2+2a_ix+x^2$ 时，预处理 $a_i$ 的前缀和可能是负数，取模后应该变成正数。