棒球遊戲II

 ins = '''

5 1B 1B FO GO 1B

5 1B 2B FO FO SO

4 SO HR SO 1B

4 FO FO FO HR

4 1B 1B 1B 1B

4 GO GO 3B GO

4 1B GO GO SO

4 SO GO 2B 2B

4 3B GO GO FO

6

'''

d1 = [ i for i in ins.strip().split('\n') ]

b = int(d1[-1])

d1.pop()

d2 = [[j for j in i.split()] for i in d1]

d = []

for i in range(1,6):

    for j in range(9):

        if int(d2[j][0])>= i:

            d.append(d2[j][i])

def aB(n):

    global score

    p.append(1)

    for i in range(n-1):

        p.append(0)

       

    for i in range(n):

        c = p.pop(0)    

        score = score + c        

def hr():

    global score

    global p

    score = score + sum(p)+1

    p=[0,0,0]

# main

score = 0

p = [0,0,0]

out = 0

bc = 0

for i in range(len(d)):

    cur = d.pop(0)

    if cur == '1B': aB(1)

    if cur == '2B': aB(2)

    if cur == '3B': aB(3)

    if cur == 'HR': hr()

    if cur[1]=='O':

        out+=1

        bc+=1

   

    if out ==3 :

        out==0

        p = [0,0,0]  

    if b == bc: break

print(score)