結果

問題 No.840 ほむほむほむら
コンテスト
ユーザー Kiona1018
提出日時 2019-09-12 00:46:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 375 ms / 4,000 ms
コード長 1,528 bytes
コンパイル時間 1,804 ms
コンパイル使用メモリ 173,912 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-02 16:57:07
合計ジャッジ時間 4,644 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

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