备忘:构造单位矩阵 ,多加 m=n
如下:
struct mp { int n, m; long long a[N][M]; void init(int row, int col, bool isI) { n=row, m=col; memset(a, 0, sizeof a); if (isI) { m=n; for (int i=1; i<=row; i++) a[i][i]=1; } } }A, B, ans;