省略
省略
C++に変換後のコードはこちら
#define MD 998244353
int N, K;
{
int i, j, k, x;
mint res;
Matrix<mint> m;
rd(N,K);
m.changeSize(K**3, K**3);
m = 0;
rep(i,K) rep(j,K) rep(k,K){
x = b[K](i,j,k);
m[x][ b[K](i,j,(k+1)%K) ] += 1;
m[x][ b[K](i,(j+k)%K,k) ] += 1;
m[x][ b[K]((i+j)%K,j,k) ] += 1;
}
m **= N;
res = 0;
rep(j,K) rep(k,K) res += m[0][ b[K](0,j,k) ];
wt(res);
}
Current time: 2024年05月03日10時08分38秒
Last modified: 2019年06月28日03時10分04秒 (by laycrs)
Tags: Competitive_Programming_Incomplete yukicoder
トップページに戻る
Logged in as: unknown user (not login)