[프로그래머스/PYTHON] 크레인 인형뽑기 게임

● 문제 링크

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

---

● 문제 설명

 

문제 설명이 길어서 위 문제 링크로 확인 부탁드립니다!

---

● 문제 해결 방향

초반에는 이중 배열로 되어있어 행과 열을 서로 바꾸어 주는 방법도 생각했지만

이번 문제의 경우 바꾸어 주지 않아도 충분히 해결 가능해 보였습니다.

 

크레인이 내려가는 열에 대해서 0이 아닌 숫자를 만날 때까지 반복하고

basket 배열의 마지막 값과 비교하여 적절한 연산을 해주었습니다.

 

---

● 문제 해결 코드

def solution(board, moves):
    answer = 0
    basket = []
    
    for m in moves:
        for row in board:
            if row[m - 1] != 0:
                if (basket == []) or (row[m - 1] != basket[-1]):
                    basket.append(row[m - 1])
                else :
                    basket = basket[:-1]
                    answer += 2
                break
        
        
        row[m - 1] = 0
    return answer

 

---

● 알게 된 것 / 아쉬운 점