AtcoderのRuntimeErrorを解消したい [Python]

Question

初めてAtcoderに参加して、こちらの問題を解いていたところ、ずっとREから変わらず解けなかったため理由を教えていただきたいです。

私の回答は以下のようになっています。

n = int(input())
list_x = []
for i in range(n):
    array = list(map(int, input().split()))
    list_x.append(array)
a = list_x[0]
b = list_x[1]
c = list_x[2]
nlist = []
for c_i in c:
    nlist.append(b[c_i-1])
ans = 0
for a_i in a:
    ans = ans + nlist.count(a_i)
print(str(ans))

どこでエラーが発生しているのでしょうか。
よろしくお願いします。

Answer 1

n = int(input())
for i in range(n):
    array = list(map(int, input().split()))
    list_x.append(array)

問題の最初の入力で受け取るNは、「A,B,Cの要素数」です。
あくまで入力そのものはA,B,Cの3回分行えば十分です。

このコードだと、Nに応じで入力待ち回数が変動しているため、
出力例2の時点ですでに処理が進まずREになってると思います。
（N=4なので、本来不要な4回目のinputが動いている）

Comments

[WatanabeJin]

そういうことですね。
ローカルの場合は4でも動いていたのが原因でした。
お早い回答ありがとうございます。