Atcoder(ABC173-C)の問題で実行エラーになってしまう理由がわかりません

Question

前提・実現したいこと

Atcoder(ABC173-C)が実行時エラー(RE)になってしまいます。
https://atcoder.jp/contests/abc173/tasks

.を0、#を1にするように置き換えて、H×Wの行列を作りました。
縦と横を消すという処理をH×W行列の前後にH×HとW×W行列をかけることによって表現することを考えました。
H×HとW×W行列は0と1の組合せが記述されている対角行列です。
計算後の行列の1の個数を数えることによって答えを出力します。

どなたか、実行時エラーになってしまう理由を知りたいです。よろしくお願いします。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

import numpy as np
import itertools
 
H, W, K = list(map(int, input().split())) 
c = [input().split() for l in range(H)]
b=[]
d=[]
for m in range(len(c)):
  for item in c[m]:
    item_mod = item.replace("."  , "0")
    item_mod = item_mod.replace("#" , "1")
    b.append(item_mod)
  d.append(b)
  b=[]
 
C_arr = np.array(d , dtype = "int")
 
In = np.array(list(itertools.product([0,1] , repeat=H)))
Out = np.array(list(itertools.product([0,1] , repeat=W)))
 
ans = 0
for i in range(2**H):
  for j in range(2**W):
    In_matrix = np.diag(In[i])
    Out_matrix = np.diag(Out[j])
    
    In_C = np.dot(In_matrix , C_arr)
    In_C_Out = np.dot(In_C , Out_matrix)
    
    ICO = In_C_Out.flatten()
    ICO = ICO.tolist()
    
    if ICO.count(1) == K:
      ans += 1
      
print(ans)

試したこと

入力例を一通り試したところ、うまくいきました。
atcoderでも数問は正解になりました

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[can110]

質問コードをそのままAtcoderに提出しましたが、全問不正解でした。
実際に実行したコードと質問に記載しているコードに相違がないか確認ください。

[nontan_1993]

理由は全く分からないのですが、
Python(3.82)にするとb05とb14がACで、それ以外がREになります
PyPy2,3だと全問REになるようです。

[can110]

> 入力例を一通り試したところ、うまくいきました。
とありますが、問題文に記載されている４つの例はうまくいったということでしょうか？
こちらのローカルマシン環境(Python3.7)ではランタイムエラーが発生しますが。

[nontan_1993]

これが正しいやり方なのかは分からないのですが、
入力例4
H,W,K = 6,6,8
c=[[".",".","#","#",".","."],[".","#",".",".","#","."],["#",".",".",".",".","#"],["#","#","#","#","#","#"],["#",".",".",".",".","#"],["#",".",".",".",".","#"]]
を入れて動かすと、答え208が出ます。
これらの作業をそれぞれ行ったら、答えが一致したという意味で記入しました。

[can110]

なるほど。了解です。

Answer 1

C_arr = np.array(d , dtype = "int")の後に以下のコードを追加してローカル環境で例題を実行してみてください。

print(C_arr)
print(C_arr.shape)

おそらく意図した配列にはなっていないと思います。
原因は入力文字列から配列作成の部分のコードに問題があるからです。
適切に修正すれば通ります。

Comments

[nontan_1993]

初心者質問で申し訳ないのですが、
H,W,K = 2,3,4
c=[[".",".","#"],["#","#","#"]]
を入れると
print(C_arr)
[[0 0 1]
[1 1 1]]
print(C_arr.shape)
(2, 3)
となり、結局、欲しい2×3(H×W)の行列になっているので、何が問題なのかが分からないのですが...

[can110]

AtCoderでは問題は標準入力から文字列として与えられます。
すなわちこの文字列から「c=[[".",".","#"],["#","#","#"]]」を作りだす処理が必要なわけですが
その処理に問題があるのでREとなっているということです。

[nontan_1993]

分かりました。ありがとうございます。