[알고리즘] Programmers [1차] 프렌즈 4블록 #Python

[Programmers] [1차] 프렌즈 4블록 바로가기

📍 문제 설명

프렌즈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가 사용된다.

---

📍 출력 형식

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

---

📍 입출력 예제

m	n		board
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개의 블록이 지워진다.

---

📍 문제 풀이

• 현재 게임판(B)에 존재하는 2X2 크기의 같은 모양의 카카오 프렌즈 블록의 좌측상단의 블록 위치를 points 배열에 저장한다.
  • 만약 2X2 크기의 같은 모양의 카카오 프렌즈 블록을 발견 즉시 제거한다면 블록이 공유되는 다른 경우를 확인할 수 없으므로 즉시 제거하지 않고 좌측 상단의 블록 위치를 저장한다.
    
• 게임판(B)에서 제거해야할 2X2 크기의 같은 모양의 카카오 프렌즈 블록을 points 배열에서 확인한 후 모두 제거한다.
• 블록 제거로 인해 발생한 빈 공간을 채운 후 위의 방식을 반복한다.
• 반복중 블록 제거가 발생하지 않았을 경우 반복을 종료한다.
• 2X2 크기의 같은 모양의 카카오 프렌즈 블록을 제거할 수 없는 상태인 게임판(B)에서 제거된 블록(0)의 수를 count() 하여 반환한다.

✍ 코드

# 게임판에 존재하는 2X2 크기의 같은 모양의 카카오 프렌즈 블록 제거하는 함수
def delSquare(m,n,B):
    points = []
    for y in range(m-1):
        for x in range(n-1):
            if B[y][x] == 0: continue
            if B[y][x] == B[y+1][x] and B[y][x] == B[y][x+1] and B[y][x] == B[y+1][x+1]:
                points.append([x,y])
    if not points:
        return B, False
    for x, y in points:
        B[y][x], B[y+1][x], B[y][x+1], B[y+1][x+1] = 0, 0, 0, 0
    B = fillEmptySpace(m,n,B)
    return B, True

# 카카오 프렌즈 블록을 제거한 후 빈 칸을 채우는 함수
def fillEmptySpace(m,n,B):
    for y in range(m-2,-1,-1):
        for x in range(n):
            if B[y][x] != 0 and B[y+1][x] == 0:
                ny = y + 1
                while ny < m and B[ny][x] == 0:
                    ny += 1
                B[y][x], B[ny-1][x] = 0, B[y][x]
    return B

def solution(m, n, B):
    answer = 0
    for i in range(len(B)):
        B[i] = list(B[i])
    
    while True:
        B, check = delSquare(m,n,B)
        if not check: break
    
    return sum([b.count(0) for b in B])