[프로그래머스] 프렌즈4블록 - 스택? (python) - 2018 KAKAO BLIND RECRUITMENT

1. 문제 설명

프렌즈4블록

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 프렌즈4블록.
같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.

 

만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.

 

 

블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.

 

 

만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.

 

 

위 초기 배치를 문자로 표시하면 아래와 같다.

TTTANT RRFACC RRRFCC TRRRAA TTMMMF TMMTTJ

각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다

입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라.

 

---

2. 입력 형식

• 입력으로 판의 높이 m, 폭 n과 판의 배치 정보 board가 들어온다.
• 2 ≦ n, m ≦ 30
• board는 길이 n인 문자열 m개의 배열로 주어진다. 블록을 나타내는 문자는 대문자 A에서 Z가 사용된다.

---

3. 출력 형식

입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.

 

---

4. 입출력 예

m	n	board	answer
4	5	["CCBDE", "AAADE", "AAABF", "CCBBF"]	14
6	6	["TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"]	15

 

---

5. 나의 풀이

올해 상반기 모 회사 코딩테스트 문제와 매우 흡사했다. (그땐 못풀었는데,,,,,, ༼;´༎ຶ ۝༎ຶ`༽  이제 풀 줄 알면 되는거지 뭐,,,,,,,,,) 

내 생각에 이 문제 풀이의 관건은 지워지는 블록의 자리를 위에 있는 블록으로 어떻게 채우냐는 것 같다. 약 2개월 전의 나는 이걸 함수로 구현해서 복잡도가 어마어마해졌고, 결국 시간초과로 통과하지 못했다. 곰곰히 생각하다 생각해낸 것은 리스트를 시계방향으로 한번 회전해서 풀면 pop을 했을 때 뒤에 있는 원소로 빈자리가 메꿔진다는 점이었다. 그래서 리스트를 회전시키려고 했는데.... 했는데...! 도저히 생각이 안났다. 그래서 구글링했다..^^.... 한줄로 기가막히게 리스트가 회전이 되었고 따로 삭제할 블록을 찾는 함수와 블록을 지우고 0을 새로 채워넣는 함수를 작성하고 마무리 했다.

 

def is2x2(m, n, board):
    answer = []
    for i in range(m-1):
        for j in range(n-1):
            temp = board[i][j]
            if temp == 0:
                continue
            if temp == board[i+1][j] and temp == board[i][j+1] and temp == board[i+1][j+1]:
                answer += [[i, j], [i+1, j], [i, j+1], [i+1, j+1]]
                
    return list(set(map(tuple, answer))) #중복된 원소를 삭제하기 위해 set으로 바꿨다 리스트로 다시 변환

def removeBlock(m, board, l):
    for i in l:
        board[i[0]].pop(i[1])
    for j in range(m):
        if len(board[j]) < m:
            board[j] += [0] * (m - len(board[j]))
    return board
    
def solution(m, n, board):
    board = list(map(list, zip(*board[::-1]))) #보드를 시계방향으로 90도 회전
    m, n = n, m
    answer = 0
    while(True):
        temp = is2x2(m, n, board)
        if temp == []:
            break
        answer += len(temp)
        board = removeBlock(m, board, sorted(temp, reverse = True)) #---(*)
        
    return answer

 

(*) 부분에서 리스트를 거꾸로 정렬하고 함수에 넣은 이유는 연속된 자리를 지울 때, 앞에 있는 원소를 먼저 지울 경우 pop에 의해 인덱스가 하나씩 당겨지기 때문이다. 

ex) [0, 1, 2, 3, 4] 에서 인덱스 1과 2를 pop 할 경우 인덱스 1을 먼저 pop 하면 [0, 2, 3, 4]가 되어 다음 인덱스 2자리를 지울 때 2가 아닌 3이 지워진 [0, 2, 4]가 되어버린다.

 

여러 유형 중, 게임 관련 문제와 미로찾기나 길 문제를 제일 어려워 하는 것 같다. 취약한 유형을 찾아 푸는데 푸는 문제마다 취약해서 결국에는 다 풀고있다는 슬픈 이야기...😟

 

 

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/17679

코딩테스트 연습 - [1차] 프렌즈4블록