[Level 1] 크레인 인형뽑기 게임

문제 : https://programmers.co.kr/learn/courses/30/lessons/64061

프로그래머스

▶ 코드

def solution(board, moves):
    n = len(board)
    p = [[] for i in range(n)] # 1번~N번 세로라인 배열 
    
    for i in range(n):
        for j in range(n):
            if board[i][j]!=0:
                p[j].append(board[i][j]) # 각 N번째 라인에 담아주기

    # 각 N번째 라인의 원소 순서 뒤집기 
    for a in p:
        a.reverse()
                
    my = [] # 뽑힌것 담을 배열 
    c = 0
    for a in moves:
        if len(p[a-1]) > 0:
            if len(my)>0 and p[a-1][-1] == my[-1]:
                my = my[:-1]
                p[a-1].pop()
                c += 1
            else:
                my.append(p[a-1].pop())
    #print(my)
    
    return c*2