yukicoder No.840 - ほむほむほむら

Source

ニコニコミュニティ
問題文

問題概要

省略

解法

省略

cLay（version 20190626-1）のコード

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年04月20日00時29分51秒
Last modified: 2019年06月28日03時10分04秒 (by laycrs)
Tags: Competitive_Programming_Incomplete yukicoder
トップページに戻る

Logged in as: unknown user (not login)