結果

問題 No.1050 Zero (Maximum)
コンテスト
ユーザー mkawa2
提出日時 2020-05-09 18:07:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 479 bytes
コンパイル時間 115 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 44,720 KB
最終ジャッジ日時 2024-07-06 00:09:11
合計ジャッジ時間 11,498 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
import sys

def MI(): return map(int, sys.stdin.readline().split())

def main():
    md=10**9+7
    m,k=MI()
    base=np.zeros((m,m),dtype=np.object)
    for i in range(m):
        for a in range(m):
            j=(a+i)%m
            base[i][j]+=1
            j=(a*i)%m
            base[i][j]+=1

    mat=np.eye(m,dtype=np.object)
    while k:
        if k&1:mat=np.dot(mat,base)%md
        base=np.dot(base,base)%md
        k>>=1
    print(mat[0][0])

main()
0