注意到该篇题解在第 $47$ 行有如下代码：

rep(i, 1, tt) to[col[i]] = i % k;

而 col[i] 可能为 $-1$，其实际上会 RE。

在洛谷上貌似不会出现问题，然而在本地或者其他部分平台上会寄