python AtCoder Bingoについて(ABC157_B)

Question

AtCoderでBingoを解いています。
以下のコードで提出したところ正解とならなかったのですが、原因が分からないので教えていただけると助かります。。

#ビンゴのA11-A33は一列にしてaとする。
#Bの中身と一致する数字がaにある場合はその箇所を0で置き換える。
#縦、横、斜めで0が3つ続いている箇所がある場合はYesと出力する。
#なければNoと出力する。
a1=[int(x) for x in input().split()]
a2=[int(x) for x in input().split()]
a3=[int(x) for x in input().split()]
n = int(input())
b=[0]*n
for i in range(n):
  b[i] = int(input())
a = a1+a2+a3

for i in b:
  for x in range(9):
    if i == a[x]:
      a[x]=0

if a[0] == 0:
  if a[1]+a[2]==0 or a[3]+a[6]==0 or a[4]+a[8]==0:
    print("Yes")
elif a[1]+a[4]+a[7]==0 or a[3]+a[4]+a[5]==0:
  print("Yes")
elif a[2]==0:
  if a[5]+a[8]==0 or a[4]+a[6]==0:
    print("Yes")
elif a[6]+a[7]+a[8]==0:
  print("Yes")
else:
  print("No")

解決したので、上手くいったコードも貼っておきます。

import sys
a1=[int(x) for x in input().split()]
a2=[int(x) for x in input().split()]
a3=[int(x) for x in input().split()]
n = int(input())
b=[0]*n
for i in range(n):
  b[i] = int(input())
a = a1+a2+a3

for i in b:
  for x in range(9):
    if i == a[x]:
      a[x]=0
      
if a[0] == 0:
  if a[1]+a[2]==0 or a[3]+a[6]==0 or a[4]+a[8]==0:
    print("Yes")
    sys.exit()
if a[1]+a[4]+a[7]==0 or a[3]+a[4]+a[5]==0:
  print("Yes")
  sys.exit()
if a[2]==0:
  if a[5]+a[8]==0 or a[4]+a[6]==0:
    print("Yes")
    sys.exit()
if a[6]+a[7]+a[8]==0:
  print("Yes")
  sys.exit()
print("No")

Answer 1

条件分岐によっては何も出力されないパターンがあります。

Comments

[kitsune_42]

ありがとうございます、お恥ずかしいです。修正して提出してみます。

[kitsune_42]

以下のようなコードにしたところ上手くいきました。
import sys
a1=[int(x) for x in input().split()]
a2=[int(x) for x in input().split()]
a3=[int(x) for x in input().split()]
n = int(input())
b=[0]*n
for i in range(n):
b[i] = int(input())
a = a1+a2+a3

for i in b:
for x in range(9):
if i == a[x]:
a[x]=0

if a[0] == 0:
if a[1]+a[2]==0 or a[3]+a[6]==0 or a[4]+a[8]==0:
print("Yes")
sys.exit()
if a[1]+a[4]+a[7]==0 or a[3]+a[4]+a[5]==0:
print("Yes")
sys.exit()
if a[2]==0:
if a[5]+a[8]==0 or a[4]+a[6]==0:
print("Yes")
sys.exit()
if a[6]+a[7]+a[8]==0:
print("Yes")
sys.exit()
print("No")