AOJ - Runtimeerrorの原因が分からない

Question

AOJの
Matrix Vector Multiplication

からです。下記コードにてruntimeerrorとなってしまうのですが、Judge Input #1とJudge Output #1の結果が一致しています。入力と出力の形式に着目してもおかしな点は見つからなかったです。Hintに「forループを使用しろ」と書いてあったのですが、numpy配列を使用した点がまずかったのでしょうか(そうは思えませんが、、、)また、aoj側のnumpyのバージョンが原因とは考えられますか？
どなたかエラーの原因をご教授いただけないでしょうか？

import numpy as np
n, m = map(int, input().split())

M = []
for i in range(1, n+1):
    row = list(map(int, input().split()))
    M.append(row)
M = np.array(M)

v = []
for i in range(1, m+1):
    v.append(int(input()))
v = np.array(v)

ans = np.dot(M, v)

for i in range(len(ans)):
    print(ans[i])

3 4
1 2 0 1
0 3 0 1
4 1 1 0
1
2
3
0

5
6
9

Answer 1

AOJ は numpy が使えません。
numpy.dot で行っている行列積の計算は自分で実装しましょう。

Comments

[esklia]

そうなんですか…importするとaoj側が大変だからですかね…？(ライブラリとかたくさんあるので)、ちなみになんですが、そのような注意書きってどこに書いてあるのでしょうか？

[tiitoi]

> importするとaoj側が大変だからですかね…？

標準ライブラリ以外を入れるとなると、沢山あるライブラリの中でどれを入れるかの線引きが難しいので、標準ライブラリ以外は使えないとしているのだと思います。

> そのような注意書きってどこに書いてあるのでしょうか？

注意書きは見つけられませんでしたが、適当な問題で Accept した解答に import numpy を1行追加したら Runtime Error となったので、import に失敗してエラーとなったと判断しました。

[esklia]

そういうことでしたか。私もご指摘通り直した後にRuntimeErrorとなり、「import numpy」を消したら成功したのでそのようですね。
丁寧に教えていただきどうもありがとうございました。