Problem p02692 (python)

Fixed submission:
n,a,b,c=map(int,input().split())
k=[]
if a+b+c==0:
  print('No')
elif a+b+c==1:
  if a==1:
    current=0
  if b==1:
    current=1
  if c==1:
    current=2
  breakflag=0
  for i in range(n):
    s=input()
    if s=='AB':
      if current==2:
        breakflag=1
      else:
        current=1-current
        k.append(current)
    if s=='AC':
      if current==1:
        breakflag=1
      else:
        current=2-current
        k.append(current)
    if s=='BC':
      if current==0:
        breakflag=1
      else:
        current=3-current
        k.append(current)
    if breakflag==1:
      print('No')
      break
    if i==n-1:
      print('Yes')
      for j in range(n):
        if k[j]==0:
          print('A')
        elif k[j]==1:
          print('B')
        else:
          print('C')
elif a+b+c==2:
  s=input()
  breakflag=0
  for i in range(n-1):
    s2=input()
    if s=='AB':
      if a==0 and b==0:
        breakflag=1
      elif a>b:
        a-=1
        b+=1
        k.append('B')
      elif b>a:
        b-=1
        a+=1
        k.append('A')
      else:
        if s2=='AC':
          a+=1
          b-=1
          k.append('A')
        else:
          a-=1
          b+=1
          k.append('B')
    elif s=='AC':
      if a==0 and c==0:
        breakflag=1
      elif a>c:
        a-=1
        c+=1
        k.append('C')
      elif c>a:
        c-=1
        a+=1
        k.append('A')
      else:
        if s2=='AB':
          a+=1
          c-=1
          k.append('A')
        else:
          a-=1
          c+=1
          k.append('C')
    elif s=='BC':
      if b==0 and c==0:
        breakflag=1
      elif b>c:
        b-=1
        c+=1
        k.append('C')
      elif c>b:
        c-=1
        b+=1
        k.append('B')
      else:
        if s2=='AB':
          b+=1
          c-=1
          k.append('B')
        else:
          b-=1
          c+=1
          k.append('C')
    if breakflag==1:
      print('No')
      break
    s=s2
  if breakflag==0:
    if s=='AB':
      if a==0 and b==0:
        print('No')
        breakflag=1
      elif a>=b:
        k.append('B')
      else:
        k.append('A')
    elif s=='AC':
      if a==0 and c==0:
        print('No')
        breakflag=1
      elif a>=c:
        k.append('C')
      else:
        k.append('A')
    else:
      if b==0 and c==0:
        print('No')
        breakflag=1
      elif b>=c:
        k.append('C')
      else:
        k.append('B')  
  if breakflag==0:
    print('Yes')
    for i in k:
      print(i)
else:
  breakflag=0
  for i in range(n):
    s=input()
    if s=='AB':
      if a==0 and b==0:
        breakflag=1
      elif a<b:
        a+=1
        b-=1
        k.append('A')
      else:
        b+=1
        a-=1
        k.append('B')
    if s=='AC':
      if a==0 and c==0:
        breakflag=1
      elif a<c:
        a+=1
        c-=1
        k.append('A')
      else:
        c+=1
        a-=1
        k.append('C')
    if s=='BC':
      if b==0 and c==0:
        breakflag=1
      elif b<c:
        b+=1
        c-=1
        k.append('B')
      else:
        c+=1
        b-=1
        k.append('C')
    if breakflag==1:
      print('No')
      break
  if breakflag==0:
    print('Yes')
    for i in k:
      print(i)

Buggy submission (verdict: Runtime Error):
n,a,b,c=map(int,input().split())
k=[]
if a+b+c==0:
  print('No')
elif a+b+c==1:
  if a==1:
    current=0
  if b==1:
    current=1
  if c==1:
    current=2
  breakflag=0
  for i in range(n):
    s=input()
    if s=='AB':
      if current==2:
        breakflag=1
      else:
        current=1-current
        k.append(current)
    if s=='AC':
      if current==1:
        breakflag=1
      else:
        current=2-current
        k.append(current)
    if s=='BC':
      if current==0:
        breakflag=1
      else:
        current=3-current
        k.append(current)
    if breakflag==1:
      print('No')
      break
    if i==n-1:
      print('Yes')
      for j in range(n):
        if k[j]==0:
          print('A')
        elif k[j]==1:
          print('B')
        else:
          print('C')
elif a+b+c==2:
  s=input()
  for i in range(n-1):
    s2=input()
    if s=='AB':
      if a==0 and b==0:
        breakflag=1
      elif a>b:
        a-=1
        b+=1
        k.append('A')
      elif b>a:
        b-=1
        a+=1
        k.append('B')
      else:
        if s2=='AC':
          a+=1
          b-=1
          k.append('A')
        else:
          a-=1
          b+=1
          k.append('B')
    elif s=='AC':
      if a==0 and c==0:
        breakflag=1
      elif a>c:
        a-=1
        c+=1
        k.append('A')
      elif c>a:
        c-=1
        a+=1
        k.append('C')
      else:
        if s2=='AB':
          a+=1
          c-=1
          k.append('A')
        else:
          a-=1
          c+=1
          k.append('C')
    elif s=='BC':
      if b==0 and c==0:
        breakflag=1
      elif b>c:
        b-=1
        c+=1
        k.append('B')
      elif c>b:
        c-=1
        b+=1
        k.append('C')
      else:
        if s2=='AB':
          b+=1
          c-=1
          k.append('B')
        else:
          b-=1
          c+=1
          k.append('C')
    if breakflag==1:
      print('No')
      break
    s=s2.copy()
  if breakflag==0:
    if s=='AB':
      if a==0 and b==0:
        print('No')
        breakflag=1
      elif a>=b:
        k.append('A')
      else:
        k.append('B')
    elif s=='AC':
      if a==0 and c==0:
        print('No')
        breakflag=1
      elif a>=c:
        k.append('A')
      else:
        k.append('C')
    else:
      if b==0 and c==0:
        print('No')
        breakflag=1
      elif b>=c:
        k.append('B')
      else:
        k.append('C')  
  if breakflag==0:
    print('Yes')
    for i in k:
      print(i)
else:
  breakflag=0
  for i in range(n):
    s=input()
    if s=='AB':
      if a==0 and b==0:
        breakflag=1
      elif a<b:
        a+=1
        b-=1
        k.append('A')
      else:
        b+=1
        a-=1
        k.append('B')
    if s=='AC':
      if a==0 and c==0:
        breakflag=1
      elif a<c:
        a+=1
        c-=1
        k.append('A')
      else:
        c+=1
        a-=1
        k.append('C')
    if s=='BC':
      if b==0 and c==0:
        breakflag=1
      elif b<c:
        b+=1
        c-=1
        k.append('B')
      else:
        c+=1
        b-=1
        k.append('C')
    if breakflag==1:
      print('No')
      break
  if breakflag==0:
    print('Yes')
    for i in k:
      print(i)