結果

問題 No.3045 反復重み付き累積和
ユーザー titia
提出日時 2025-03-06 02:58:24
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 925 bytes
コンパイル時間 439 ms
コンパイル使用メモリ 82,640 KB
実行使用メモリ 402,756 KB
最終ジャッジ日時 2025-03-06 02:58:32
合計ジャッジ時間 7,741 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 TLE * 1 -- * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

def prod(A,B,k,l,m):# A:k*l,B:l*m
    C=[[0 for i in range(m)] for j in range(k)]

    for i in range(k):
        for j in range(m):
            ANS=0
            for pl in range(l):
                ANS=(ANS+A[i][pl]*B[pl][j])%mod

            C[i][j]=ANS

    return C

N,Q=map(int,input().split())
A=list(map(int,input().split()))
mod=998244353

A=[A]
for tests in range(Q):
    L=list(map(int,input().split()))

    if L[0]==2:
        x=L[1]-1

        print(A[0][x])
        continue

    k,x=L[1],L[2]

    L=[1]
    bunshi=x
    bunbo=1

    for i in range(1,N+1):
        now=L[-1]
        L.append(now*k*bunshi*pow(bunbo,mod-2,mod)%mod)
        bunshi+=1
        bunbo+=1

    X=[[0]*N for i in range(N)]

    for i in range(N):
        for j in range(N):
            if i+j<N:
                X[i][i+j]=L[j]
            else:
                break

    A=prod(A,X,1,N,N)
0