結果

問題 No.840 ほむほむほむら
ユーザー maspymaspy
提出日時 2020-01-02 16:48:46
言語 Python3
(3.11.6 + numpy 1.26.0 + scipy 1.11.3)
結果
AC  
実行時間 369 ms / 4,000 ms
コード長 1,438 bytes
コンパイル時間 254 ms
コンパイル使用メモリ 10,824 KB
実行使用メモリ 31,216 KB
最終ジャッジ日時 2023-08-14 18:33:10
合計ジャッジ時間 6,753 ms
ジャッジサーバーID
(参考情報)
judge14 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 109 ms
29,720 KB
testcase_01 AC 108 ms
29,808 KB
testcase_02 AC 113 ms
29,924 KB
testcase_03 AC 139 ms
30,956 KB
testcase_04 AC 105 ms
29,744 KB
testcase_05 AC 107 ms
29,640 KB
testcase_06 AC 109 ms
29,752 KB
testcase_07 AC 120 ms
29,952 KB
testcase_08 AC 169 ms
31,204 KB
testcase_09 AC 109 ms
29,848 KB
testcase_10 AC 109 ms
29,768 KB
testcase_11 AC 109 ms
29,652 KB
testcase_12 AC 125 ms
29,928 KB
testcase_13 AC 268 ms
31,008 KB
testcase_14 AC 131 ms
30,060 KB
testcase_15 AC 117 ms
29,788 KB
testcase_16 AC 118 ms
29,656 KB
testcase_17 AC 150 ms
30,160 KB
testcase_18 AC 320 ms
31,096 KB
testcase_19 AC 369 ms
31,216 KB
testcase_20 AC 124 ms
29,688 KB
testcase_21 AC 110 ms
29,748 KB
testcase_22 AC 111 ms
30,044 KB
testcase_23 AC 343 ms
31,116 KB
testcase_24 AC 114 ms
29,736 KB
testcase_25 AC 107 ms
29,828 KB
testcase_26 AC 112 ms
29,912 KB
testcase_27 AC 345 ms
31,152 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

import itertools
import numpy as np

"""
・empty, ほ、ほむ、ほむら の個数 mod K の分布ごとに、文字列数を持つ。size <= 125の行列
・遷移は行列で書ける
"""

N,K = map(int,read().split())

MOD = 998244353

def get_index(ho,homu,homura):
    return (ho + homu * K) * K + homura

def create_mat(K):
    L = K ** 3
    A = np.zeros((L,L),np.int64)
    for a,b,c in itertools.product(range(K), repeat = 3):
        i = get_index(a,b,c)
        # ほ
        j = get_index((a+1)%K,b,c)
        A[j,i] += 1
        # む
        j = get_index(a,(a+b)%K,c)
        A[j,i] += 1
        # ら
        j = get_index(a,b,(b+c)%K)
        A[j,i] += 1
    return A

def mult(A,B,MOD):
    mask = (1 << 15) - 1
    A1 = A >> 15; A2 = A & mask
    B1 = B >> 15; B2 = B & mask
    X = np.dot(A1,B1) % MOD
    Y = np.dot(A2,B2) % MOD
    Z = (np.dot(A1+A2,B1+B2) - X - Y) % MOD
    C = (X << 30) + (Z << 15) + Y
    return C % MOD

def power_mat(A,n,MOD):
    k = A.shape[0]
    if n == 0:
        return np.eye(k,dtype=np.int64)
    B = power_mat(A,n//2,MOD)
    B = mult(B,B,MOD)
    return mult(A,B,MOD) if n & 1 else B

A = create_mat(K)
B = power_mat(A,N,MOD)

answer = 0
for a,b in itertools.product(range(K),repeat=2):
    i = get_index(a,b,0)
    answer += B[i,0]
answer %= MOD

print(answer)
0