結果

問題 No.2396 等差二項展開
コンテスト
ユーザー tassei903
提出日時 2023-07-28 22:20:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,894 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 191 ms
コンパイル使用メモリ 82,472 KB
実行使用メモリ 74,688 KB
最終ジャッジ日時 2024-10-06 19:27:44
合計ジャッジ時間 8,209 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
#2D matrix

def E(n):
    A = [[0 for j in range(n)]for i in range(n)]
    for i in range(n):
        A[i][i] = 1
    return A

def add(x,y):
    return (x + y)%mod

def mul(x, y):
    return x * y % mod


def mat_add(A, B, replace=False):
    assert len(A)==len(B) and len(A[0]) == len(B[0])
    if not replace:
        A = [a.copy() for a in A]
    n = len(A)
    m = len(A[0])
    for i in range(n):
        for j in range(m):
            A[i][j] = add(A[i][j], B[i][j])
    return A


def mat_mul(A,B):
    assert len(A[0]) == len(B)
    n = len(A)
    m = len(B[0])
    p = len(A[0])
    R = [[0 for j in range(m)]for i in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(p):
                R[i][j] = add(R[i][j], mul(A[i][k],B[k][j]))
    return R

def mat_pow(A, x):
    assert len(A)==len(A[0])
    n = len(A)
    R = E(n)
    while x > 0:
        if x&1:
            R = mat_mul(R, A)
        A = mat_mul(A,A)
        x >>= 1
    return R



def mat_pri(A):
    for i in A:
        print(*i)
n,m,l,k,b = na()
mod = l * b

a = [0] * l
a[0] = 1 % b
a[1 % l] = m % b
f = [0] * l
f[0] = 1
#print(f, a)
while n:
    if n % 2:
        nf = [0] * l
        for i in range(l):
            for j in range(l):
                nf[(i+j)%l] += f[i] * a[j] % b
                nf[(i+j)%l] %= b
        f = nf
    na = [0] * l
    for i in range(l):
        for j in range(l):
            na[(i+j)%l] += a[i] * a[j] % b
            na[(i+j)%l] %= b
    a = na
    #print(f, a)
    n//=2
#print(f)
print(f[k])
0