[ Programmers / Coding Test / Python ] 프렌즈4블록

황승환·2022년 6월 7일
0

Python

목록 보기
323/498

문제 설명

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈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)을 의미한다

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

입력 형식

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

출력 형식

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

입출력 예제

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

예제에 대한 설명

입출력 예제 1의 경우, 첫 번째에는 A 블록 6개가 지워지고, 두 번째에는 B 블록 4개와 C 블록 4개가 지워져, 모두 14개의 블록이 지워진다.
입출력 예제 2는 본문 설명에 있는 그림을 옮긴 것이다. 11개와 4개의 블록이 차례로 지워지며, 모두 15개의 블록이 지워진다.

접근 방법

이번 문제는 문자열을 리스트로 변환한 후에, 브루트포스로 모든 점을 순회하며 해당 점에서 오른쪽, 아래, 오른쪽아래의 위치를 확인하여 현재 점과 값이 같을 경우에 이들을 없애는 문제였다. 여기서 중요한 것은 한번의 순회에 지워지는 블록들은 모두 동시에 지워져야 한다는 것이다. 그래서 임시 리스트를 False로 채워 선언하고, 지워지는 인덱스의 값만 True로 변환한 후에, 임시 리스트를 순회하며 False 일 때만 새로운 리스트에 값을 옮긴다. 아래로 쌓이는 형태이므로, 임시 변수 idx를 사용하여 하나의 값을 쌓을 때만 idx를 감소시키는 방법으로 하여 위에서부터 아래로 쌓는다.

이 과정을 board가 함수 호출 이전과 같을 때까지 반복한 후에, board를 순회하며 .의 갯수를 answer로 반환하도록 하였다.

solution.py

import copy
def crash(m, n, board):
    dy, dx=[0, 1, 1], [1, 0, 1]
    chk=[[False for _ in range(n)] for _ in range(m)]
    for i in range(m):
        for j in range(n):
            cnt=0
            for d in range(3):
                ni, nj=i+dy[d], j+dx[d]
                if 0<=ni<m and 0<=nj<n and board[i][j]==board[ni][nj]:
                    cnt+=1
            if cnt==3:
                chk[i][j]=True
                for d in range(3):
                    ni, nj=i+dy[d], j+dx[d]
                    chk[ni][nj]=True
    result=[['' for _ in range(n)] for _ in range(m)]
    for i in range(n):
        idx=m-1
        for j in range(m-1, -1, -1):
            if not chk[j][i]:
                result[idx][i]=board[j][i]
                idx-=1
    return result
def solution(m, n, board):
    answer=0
    for i in range(m):
        board[i]=list(str(board[i]))
    while True:
        tmp=copy.deepcopy(board)
        board=copy.deepcopy(crash(m, n, board))
        if tmp==board:
            for i in range(m):
                answer+=board[i].count('')
            break
    return answer

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글