[프로그래머스] 크레인 인형뽑기 게임 (python)

문제링크

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

코딩테스트 연습 - 크레인 인형뽑기 게임

---

내풀이

import numpy as np
from collections import deque

def solution(board, moves):
    b = np.array(board).T.tolist()
    for doll in b:
        for i in range(len(b[0])):
            try:
                doll.remove(0)
            except:
                continue
    answer, s = 0, []
    for m in moves:
        if len(b[m-1]) >= 1:
            s.append(b[m-1][0])
            del b[m-1][0]

    before_len = len(s)
    before = ''
    while s != before:
        before = s[:]
        for i in range(len(s) - 1):
            if s[i] == s[i+1]:
                del s[i]
                del s[i]
                break

    return before_len - len(s)

---

피드백

2중 반복문으로 더 깔끔히 짤 수 있음