「【行によって長さが違う二次元配列の表示】すべての行の長さと値が不定な 2 次元配列の出力」のFINAL問題の解答例

目次

はじめに
回答例

はじめに

最近、paizaでプログラミング(Python)の勉強(おさらいとスキルアップ)をしているのですが、「【行によって長さが違う二次元配列の表示】すべての行の長さと値が不定な 2 次元配列の出力」のFINAL問題が難しかったので、備忘録として残しときます。

回答例

こちらが回答例です。

解答例

n,m = input().split(" ")
a = input().split(" ")
b = input().split(" ")

count = 0

for i in range(int(m)):
	for j in range(int(b[int(i)])):
		if j < int(b[int(i)])-1:
			print(a[j+count],end=" ")
		else:
			print(a[j+count])
			count += int(b[int(i)])