如果你TLE的话，考虑一下是不是没有特判n=0的情况。

如果WA，可能是矩阵运算出现了负数，取模运算要注意。