結果

問題 No.840 ほむほむほむら
ユーザー Kiona1018
提出日時 2019-09-12 00:32:18
言語 C++14
(gcc 8.3.0)
結果
WA   .
実行時間 -
コード長 1,509 Byte
コンパイル時間 1,453 ms
使用メモリ 1,836 KB
最終ジャッジ日時 2019-09-12 00:32:22

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
0_small01.txt AC 3 ms
1,528 KB
0_small02.txt AC 3 ms
1,556 KB
0_small03.txt AC 10 ms
1,628 KB
0_small04.txt AC 54 ms
1,808 KB
0_small05.txt AC 3 ms
1,528 KB
0_small06.txt WA -
0_small07.txt WA -
0_small08.txt WA -
0_small09.txt WA -
0_small10.txt WA -
1_large01.txt WA -
1_large02.txt WA -
1_large03.txt WA -
1_large04.txt WA -
1_large05.txt WA -
1_large06.txt WA -
1_large07.txt WA -
1_large08.txt WA -
1_large09.txt WA -
1_large10.txt WA -
2_corner01.txt AC 2 ms
1,528 KB
2_corner02.txt AC 4 ms
1,556 KB
2_corner03.txt AC 9 ms
1,628 KB
2_corner04.txt WA -
2_corner05.txt WA -
3_sample01.txt AC 3 ms
1,528 KB
3_sample02.txt AC 10 ms
1,624 KB
3_sample03.txt WA -
テストケース一括ダウンロード

ソースコード

diff #
#include<bits/stdc++.h>
#define rep(i,n,m) for(int i = (n); i <(m); i++)
#define rrep(i,n,m) for(int i = (n) - 1; i >=(m); i--)
using namespace std;
using ll = long long;
using mat = vector<vector<int>>;
const int MOD = 998244353;

mat matmul(int n, mat A, mat B)
{
    mat C(n, vector<int>(n, 0));
    rep(i, 0, n)
        rep(j, 0, n)
            rep(k, 0, n)
            {
                C[i][j] += A[i][k] * B[k][j];
                C[i][j] %= MOD;
            }
    return C;
}

int main()
{
    int N, K;
    cin >> N >> K;

    // initialize matrix
    int k3 = K * K * K;

    mat A(k3, vector<int>(k3, 0));
    rep(num, 0, k3)
    {
        int i, j, k;
        i = num % K;
        j = (num / K) % K;
        k = (num / (K * K)) % K;

        int num2 = (i + 1) % K + j * K + k * K * K;
        int num3 = i + ((i + j)%K) * K + k * K * K;
        int num4 = i + j * K + ((j + k)%K) * K * K;
        A[num2][num]++;
        A[num3][num]++;
        A[num4][num]++;
    }

    int n = N;
    mat An(k3, vector<int>(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)
    {
        int i, j, k;
        i = num % K;
        j = (num / K) % K;
        k = (num / (K * K)) % K;

        if (k % K != 0)
            continue;

        ans += An[num][0];
        ans %= MOD;

    }
    cout << ans << endl;
    return 0;
}
0