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

문제 설명

프렌즈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)을 의미한다

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

입력 형식

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

출력 형식

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

풀이

간만에 아무 해설없이 혼자 풀었다
좀 오래걸리고 주먹구구식으로 풀긴했지만 너무 뿌듯하다^^

우선 find라는 함수를 만들어서 이중포문으로 스캔을 한뒤 현재 블럭의 오른쪽, 밑, 오른쪽 밑이 다같으면
boomSet이라는 set에 추가해준다(겹치는게 있을 수 있기 때문에 set으로 함)
다 스캔하고 나면
boomSet을 활용해 터진 블럭 자리에 x 표시를 하고 그 숫자를 센다
이때 아무 블럭도 안 터졌으면 boom이 false를 리턴하고 한 무더기라도 터졌으면 true를 리턴한다.

그리고 slideDown이라는 가운데 부분이 터지고 위에 떠 있는 블럭들을 떨궈주는 함수를 만들었다.
밑에서 부터 순회하면서 현재 블럭이 x라면 그 블럭 위로 쭉 검사하여 x가 아닌 블럭이랑 바꿔치기 하는 방식으로 구현하였다.

그리고 메인 함수에는 while문으로 스캔하고 떨구고를 반복한다 폭탄이 터지지 않으면 루프를 빠져나간다.

cnt = 0
def find(m, n, board): # 스캔하여 찾은 다음 터트린 블럭을 x로 표시하는 함수
    global cnt
    boom = False # 폭발이 일어나는지 안 일어나는지 판단
    boomSet = set() # 터진 블럭들의 좌표를 저장함. 겹치는게 있을 수 있으므로 set으로 선언
    for i in range(m-1):
        for j in range(n-1):
            if board[i][j] == board[i][j+1] and board[i][j] != 'x': # 만약에 현재 블럭이 다음 블럭과 같고 x가 아니라면
                if board[i][j] == board[i+1][j]: # 근데 또 밑에 블럭도 같으면
                    if board[i][j] == board[i+1][j+1]: # 오른쪽 대각선 밑의 블럭과도 같으면
                        boomSet.add((i, j)) # 4개 블럭 모두 터진블럭 리스트에 추가하자
                        boomSet.add((i+1, j))
                        boomSet.add((i, j+1))
                        boomSet.add((i+1, j+1))
                        boom = True # 폭탄 터졌다고 체크
    for b in boomSet: # 터진 블럭들에다가 x표시하고 count 셈
        board[b[0]][b[1]] = 'x'
        cnt += 1
    return boom


def slideDown(m, n, board): # 중간에 폭탄이 터졌으면 위에 있는 블럭들을 밑으로 내려 주는 함수
    for i in range(m-1, 0, -1): # 밑에서 부터 순회하면서
        for j in range(n): # 하나씩 순회하며
            if board[i][j] == 'x': # 만약에 x이면 
                for k in range(i-1, -1, -1): # 그 위로 쭉 검사해서 x가 아닌것 나타나면 자리를 바꿈
                    if board[k][j] != 'x':
                        temp = board[k][j]
                        board[k][j] = board[i][j]
                        board[i][j] = temp
                        break


def makeBoard(board): # 입력값을 list로 만들어 주는 함수
    temp = []
    for b in board:
        temp.append(list(b))
    return temp


def solution(m, n, board):
    board = makeBoard(board) # map으로 만들어 주고
    global cnt
    boom = False 
    while 1: 
        boom = find(m, n, board) # 스캔한뒤 터진블럭들 세고 x표시함
        if not boom: # 터지지 않았다면 루프 빠져나감
            break
        slideDown(m, n, board) # 밑으로 다 떨궈줌
    return cnt