結果

問題 No.2994 べき内積
ユーザー solosolo
提出日時 2024-12-19 23:39:00
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,041 bytes
コンパイル時間 534 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 418,328 KB
最終ジャッジ日時 2024-12-19 23:40:13
合計ジャッジ時間 71,611 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
17,828 KB
testcase_01 AC 38 ms
418,328 KB
testcase_02 AC 830 ms
53,028 KB
testcase_03 TLE -
testcase_04 TLE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 TLE -
testcase_08 TLE -
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
testcase_24 TLE -
testcase_25 TLE -
権限があれば一括ダウンロードができます

ソースコード

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