結果

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

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
0_small01.txt AC 3 ms
1,532 KB
0_small02.txt AC 3 ms
1,572 KB
0_small03.txt AC 11 ms
1,668 KB
0_small04.txt AC 54 ms
2,112 KB
0_small05.txt AC 3 ms
1,532 KB
0_small06.txt AC 3 ms
1,536 KB
0_small07.txt AC 4 ms
1,572 KB
0_small08.txt AC 19 ms
1,664 KB
0_small09.txt AC 95 ms
2,112 KB
0_small10.txt AC 5 ms
1,572 KB
1_large01.txt AC 3 ms
1,536 KB
1_large02.txt AC 6 ms
1,568 KB
1_large03.txt AC 29 ms
1,664 KB
1_large04.txt AC 244 ms
2,108 KB
1_large05.txt AC 34 ms
1,668 KB
1_large06.txt AC 4 ms
1,536 KB
1_large07.txt AC 7 ms
1,568 KB
1_large08.txt AC 59 ms
1,664 KB
1_large09.txt AC 315 ms
2,108 KB
1_large10.txt AC 399 ms
2,136 KB
2_corner01.txt AC 3 ms
1,528 KB
2_corner02.txt AC 4 ms
1,572 KB
2_corner03.txt AC 8 ms
1,664 KB
2_corner04.txt AC 361 ms
2,136 KB
2_corner05.txt AC 8 ms
1,572 KB
3_sample01.txt AC 3 ms
1,536 KB
3_sample02.txt AC 11 ms
1,668 KB
3_sample03.txt AC 359 ms
2,112 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