結果

問題 No.2994 べき内積
ユーザー solosolo
提出日時 2024-12-19 23:21:49
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,343 bytes
コンパイル時間 396 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 428,928 KB
最終ジャッジ日時 2024-12-19 23:23:01
合計ジャッジ時間 71,411 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
17,828 KB
testcase_01 AC 33 ms
428,928 KB
testcase_02 AC 795 ms
53,156 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_sp = A.split()
a1_int = [int(s) for s in a1_sp]
a = [a1_int]
n_n = [i for i in range(n+1)]
#n_n.reverse()
k_kai = 0
p = 1009
amari = ''

def make_list(s):
	global 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):
	global a
	for i in range(1, k_kai):
		#for p in range(len(a[0])):だとうまくいかない
		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]
					#print(f"a[{i}][{p}] += a[{i-1}][{j}] * a[0][{n-j}] = {a[i-1][j]} * {a[0][n-j]} -> {a[i][p]}")
					#n_n.pop(0)
				n_n.pop(-1)
				n -= 1
			if n_n == []:
				n_n = [i for i in range(n+1)]
				#n_n.reverse()
				n = len(a[0])-1
			#print(a[i])
		a[i].reverse()
		#print(a[i])

def make_kai(k_kai, p):
	global amari
	#a[-1].reverse()
	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)
list_00(k_kai, len(a1_int))
#print(a[1])
make_a(n, n_n)
#print(a)
make_kai(k_kai,p)
0