結果

問題 No.2994 べき内積
ユーザー solo
提出日時 2024-12-19 23:39:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,041 bytes
コンパイル時間 534 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 418,328 KB
最終ジャッジ日時 2024-12-19 23:40:13
合計ジャッジ時間 71,611 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other TLE * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

M_N, K, A = [input() for _ in range(3)]
M_N_int = [int(s) for s in M_N.split()]
m = M_N_int.pop(0)
n = M_N_int.pop(0)
k_sp = [int(k) for k in K.split()]
a1_int = [int(s) for s in A.split()]
a = [a1_int]
n_n = [i for i in range(n+1)]
k_kai = 0
p = 1009
amari = ''

def make_list(s, a):
	for _ in range(s-1):
		a.append([0])

def list_00(s, t):
	for i in range(1, s):
		for _ in range(t-1):
			a[i].append(0)


def make_k(k):
	global k_kai
	for j in range(m+1):
		k_kai += k_sp[j]*p**j
	return k_kai

def make_a(n, n_n, a):
	for i in range(1, k_kai):
		for p in range(len(a[0])+1): 
			if n_n is not None and len(n_n)>0:
				for j in n_n:
					a[i][p] += a[i-1][j]*a[0][n-j]
				n_n.pop(-1)
				n -= 1
			if n_n == []:
				n_n = [i for i in range(n+1)]
				n = len(a[0])-1
		a[i].reverse()

def make_kai(k_kai, p):
	global amari
	for i in range(len(a1_int)):
		amari += str(a[k_kai-1][i]%p)
		if i != len(a1_int)-1:
			amari += ' '
	print(amari)

make_k(k_sp)
make_list(k_kai, a)
list_00(k_kai, len(a1_int))
make_a(n, n_n, a)
make_kai(k_kai,p)
0