結果

問題 No.840 ほむほむほむら
ユーザー Kiona1018
提出日時 2019-09-12 00:46:05
言語 C++14
(gcc 9.3.0)
結果
AC  
実行時間 396 ms
コード長 1,528 Byte
コンパイル時間 1,652 ms
使用メモリ 8,948 KB
最終ジャッジ日時 2020-04-04 23:55:05

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
scramble_00 AC 3 ms
6,900 KB
scramble_01 AC 3 ms
8,944 KB
scramble_02 AC 9 ms
8,896 KB
scramble_03 AC 53 ms
8,940 KB
scramble_04 AC 2 ms
8,948 KB
scramble_05 AC 3 ms
6,896 KB
scramble_06 AC 3 ms
6,896 KB
scramble_07 AC 18 ms
6,900 KB
scramble_08 AC 96 ms
6,900 KB
scramble_09 AC 4 ms
8,940 KB
scramble_10 AC 2 ms
8,944 KB
scramble_11 AC 5 ms
8,896 KB
scramble_12 AC 27 ms
6,900 KB
scramble_13 AC 257 ms
8,940 KB
scramble_14 AC 34 ms
8,940 KB
scramble_15 AC 2 ms
8,940 KB
scramble_16 AC 7 ms
8,944 KB
scramble_17 AC 58 ms
6,904 KB
scramble_18 AC 325 ms
8,900 KB
scramble_19 AC 396 ms
8,940 KB
scramble_20 AC 3 ms
8,896 KB
scramble_21 AC 3 ms
8,900 KB
scramble_22 AC 7 ms
8,940 KB
scramble_23 AC 375 ms
6,900 KB
scramble_24 AC 7 ms
6,900 KB
scramble_25 AC 3 ms
6,900 KB
scramble_26 AC 10 ms
8,896 KB
scramble_27 AC 377 ms
8,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
#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;
}
0