hack.in

3 3
1 0 998244353

possible answer 1996488706

correct answer 2994733058

这是因为，求出的前缀和可能有重复数字，如果您写的 trie 在叶子节点 siz[p]=1 而不是 siz[p]++，就有可能会挂。