Problem p03049 (python)

Fixed submission:
import math,string,itertools,fractions,heapq,collections,re,array,bisect,sys,random,time,queue,copy

sys.setrecursionlimit(10**7)
inf=10**20
mod=10**9+7
dd=[(-1,0),(0,1),(1,0),(0,-1)]
ddn=[(-1,0),(-1,1),(0,1),(1,1),(1,0),(1,-1),(0,-1),(-1,-1)]

def LI(): return [int(x) for x in sys.stdin.readline().split()]
def LI_(): return [int(x)-1 for x in sys.stdin.readline().split()]
def I(): return int(sys.stdin.readline())
def LS(): return sys.stdin.readline().split()
def S(): return input()

def main():
  n=I()
  ans=0

  _ba=0
  _b=0
  _a=0
  for _ in range(n):
    s=S()
    ans+=s.count('AB')

    if s[0]=='B':
      if s[-1]=='A':
        _ba+=1
      else:
        _b+=1
    else:
      if s[-1]=='A':
        _a+=1

  ans+=max(0,_ba-1)

  if _ba>0:
    if _b>0:
      ans+=1
      _b-=1
    if _a>0:
      ans+=1
      _a-=1

  if _b>0 and _a>0:
    ans+=min(_b,_a)

  return ans

# main()
print(main())

Buggy submission (verdict: Runtime Error):
n=int(input())
c1=c2=c3=0
c=0
for i in range(n):
  s=input()
  c+=s.count('AB')

  if s[0]=='B' and s[-1]=='A':
    c3+=1
  elif s[0]=='B':
    c1+=1
  elif s[-1]=='A':
    c2+=1

_c=c
_c+=max(0,c3-1)
_c1=c1
_c2=c2
if c1>0
  _c+=1
  _c1-=1
if c2>0:
  _c+=1
  _c2-=1
if c1<c2:
  _c2-=c1
  _c+=c1
  if _c2>0 and c3>0:
    _c+=1
elif c1==c2:
  if c3>0:
    _c+=1
    _c1-=1
    _c2-=1
  _c+=_c1
else:
  _c1-=c2
  _c+=c2
  if _c1>0 and c3>0:
    _c+=1

print(_c)

# if c3>max(c1,c2):
#   if c1<c2:
#     c+=c1*2
#     c3-=c1
#     c+=c3
#   elif c1==c2:
#     c+=c1*2
#     c3-=c1
#     c+=max(0,c3-1)
#   else:
#     c+=c2*2
#     c3-=c2
#     c+=c3
# elif c3>=min(c1,c2) and c3<=max(c1,c2):
#   if c1<c2:
#     c+=c1*2
#     c2-=c1
#     c3-=c1
#     c+=c3
#   else:
#     c+=c2*2
#     c1-=c2
#     c3-=c2
#     c+=c3
# else:
#   c+=c3*2
#   c1-=c3
#   c2-=c3
#   c+=min(c1,c2)

# print(max(c,_c))

Buggy submission (verdict: Runtime Error):
n=int(input())
c1=c2=c3=0
c=0
for i in range(n):
  s=input()
  c+=s.count('AB')

  if s[0]=='B' and s[-1]=='A':
    c3+=1
  elif s[0]=='B':
    c1+=1
  elif s[-1]=='A':
    c2+=1

_c=c
_c+=max(0,c3-1)
_c1=c1
_c2=c2
if c1<c2:
  _c2-=c1
  cc+=c1
  if _c2>0 and c3>0:
    cc+=1
elif c1==c2:
  cc+=c1
else:
  _c1-=c2
  cc+=c2
  if _c1>0 and c3>0:
    cc+=1

if c3>max(c1,c2):
  if c1<c2:
    c+=c1*2
    c3-=c1
    c+=c3
  elif c1==c2:
    c+=c1*2
    c3-=c1
    c+=max(0,c3-1)
  else:
    c+=c2*2
    c3-=c2
    c+=c3
elif c3>=min(c1,c2) and c3<=max(c1,c2):
  if c1<c2:
    c+=c1*2
    c2-=c1
    c3-=c1
    c+=c3
  else:
    c+=c2*2
    c1-=c2
    c3-=c2
    c+=c3
else:
  c+=c3*2
  c1-=c3
  c2-=c3
  c+=min(c1,c2)

print(c)

Buggy submission (verdict: Runtime Error):
n=int(input())
c1=c2=c3=0
c=0
for i in range(n):
  s=input()
  c+=s.count('AB')

  if s[0]=='B' and s[-1]=='A':
    c3+=1
  elif s[0]=='B':
    c1+=1
  elif s[-1]=='A':
    c2+=1

cc+=max(0,c3-1)
if c1<c2:
  c2-=c1
  cc+=c1
  if c2>0 and c3>0:
    cc+=1
elif c1==c2:
  cc+=c1
else:
  c1-=c2
  cc+=c2
  if c1>0 and c3>0:
    cc+=1

if c3>max(c1,c2):
  if c1<c2:
    c+=c1*2
    c3-=c1
    c+=c3
  elif c1==c2:
    c+=c1*2
    c3-=c1
    c+=max(0,c3-1)
  else:
    c+=c2*2
    c3-=c2
    c+=c3
elif c3>=min(c1,c2) and c3<=max(c1,c2):
  if c1<c2:
    c+=c1*2
    c2-=c1
    c3-=c1
    c+=c3
  else:
    c+=c2*2
    c1-=c2
    c3-=c2
    c+=c3
else:
  c+=c3*2
  c1-=c3
  c2-=c3
  c+=min(c1,c2)

print(max(c,cc))