結果
問題 | No.840 ほむほむほむら |
ユーザー | Kiona1018 |
提出日時 | 2019-09-12 00:46:05 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 374 ms / 4,000 ms |
コード長 | 1,528 bytes |
コンパイル時間 | 1,602 ms |
コンパイル使用メモリ | 171,192 KB |
実行使用メモリ | 4,384 KB |
最終ジャッジ日時 | 2023-09-15 14:03:50 |
合計ジャッジ時間 | 4,659 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 8 ms
4,376 KB |
testcase_03 | AC | 52 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 2 ms
4,380 KB |
testcase_06 | AC | 3 ms
4,380 KB |
testcase_07 | AC | 17 ms
4,380 KB |
testcase_08 | AC | 93 ms
4,376 KB |
testcase_09 | AC | 3 ms
4,376 KB |
testcase_10 | AC | 1 ms
4,376 KB |
testcase_11 | AC | 4 ms
4,376 KB |
testcase_12 | AC | 26 ms
4,376 KB |
testcase_13 | AC | 240 ms
4,376 KB |
testcase_14 | AC | 31 ms
4,376 KB |
testcase_15 | AC | 1 ms
4,380 KB |
testcase_16 | AC | 5 ms
4,380 KB |
testcase_17 | AC | 57 ms
4,380 KB |
testcase_18 | AC | 308 ms
4,376 KB |
testcase_19 | AC | 374 ms
4,376 KB |
testcase_20 | AC | 1 ms
4,376 KB |
testcase_21 | AC | 2 ms
4,376 KB |
testcase_22 | AC | 6 ms
4,380 KB |
testcase_23 | AC | 357 ms
4,376 KB |
testcase_24 | AC | 6 ms
4,384 KB |
testcase_25 | AC | 2 ms
4,376 KB |
testcase_26 | AC | 8 ms
4,376 KB |
testcase_27 | AC | 358 ms
4,376 KB |
ソースコード
#include<bits/stdc++.h> #define rep(i,n,m) for(ll i = (n); i <(m); i++) #define rrep(i,n,m) for(ll i = (n) - 1; i >=(m); i--) using namespace std; using ll = long long; using mat = vector<vector<ll>>; const ll MOD = 998244353; mat matmul(ll n, mat A, mat B) { mat C(n, vector<ll>(n, 0)); rep(i, 0, n) rep(j, 0, n) rep(k, 0, n) { C[i][j] += (A[i][k] * B[k][j]) % MOD; C[i][j] %= MOD; } return C; } int main() { ll N, K; cin >> N >> K; // initialize matrix ll k3 = K * K * K; mat A(k3, vector<ll>(k3, 0)); rep(num, 0, k3) { ll i, j, k; i = num % K; j = (num / K) % K; k = (num / (K * K)) % K; ll num2 = (i + 1) % K + j * K + k * K * K; ll num3 = i + ((i + j)%K) * K + k * K * K; ll num4 = i + j * K + ((j + k)%K) * K * K; A[num2][num]++; A[num3][num]++; A[num4][num]++; } ll n = N; mat An(k3, vector<ll>(k3, 0)); rep(i, 0, k3) An[i][i] = 1; while (n) { if (n & 1) { An = matmul(k3, A, An); } A = matmul(k3, A, A); n >>= 1; } ll ans = 0; rep(num, 0, k3) { ll i, j, k; i = num % K; j = (num / K) % K; k = (num / (K * K)) % K; if (k != 0) continue; ans += An[num][0]; ans %= MOD; // cout << ans << endl; } cout << ans << endl; return 0; }