[Python] 프렌즈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가 사용된다.
출력 형식
입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.
입출력 예제
mnboardanswer
| 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개의 블록이 지워진다.
풀이
아이디어
(1) 2*2 블록이 겹쳐있을 때 처리하는 방법: 2*2 블록에 해당하는 블록의 행,열을 list에 다 담고 set으로 중복 제거 처리 -> 이후 count
그리고 set에 있는 블록들 '0'으로 바꾸기
(2) 내려 가는 거 구현: for문 세 번 사용해서 아래가 비어있으면 위의 값과 바꾸기
def solution(m, n, board):
# input: m은 행의 개수, n의 열의 개수
# output: 몇 개 블록이 지워질지
answer = 0 # 총 사라진 블록 수
# board를 2차원 list로
b = [[j for j in range(n)] for i in range(m)]
for i in range(m):
for j in range(n):
b[i][j] = board[i][j]
# tip: for i in range(m): board[i] = list(board[i])
# b = [list(x) for x in board]
# issue: 어떻게 없어진 애 처리?
# 2*2 찾고 0으로 -> 0인 애들은 무시하고 2*2를 찾게
while True: # 알아서 종료될 때까지 돌림
# issue: 어떻게 2*2 블록이 겹쳐있을 때 처리? -> set(집합) 사용
pop = [] # 이번에 터져야 할 블록들 모음
for i in range(m-1):
for j in range(n-1):
if b[i][j] == b[i][j+1] == b[i+1][j] == b[i+1][j+1] != '0': # tip
pop.append((i,j))
pop.append((i+1,j))
pop.append((i,j+1))
pop.append((i+1,j+1))
if len(pop) == 0:
break
answer += len(set(pop)) # 터져야 할 블록 처리하기
for p in pop:
b[p[0]][p[1]] = '0'
# issue: 블록들 내리기-> 블록이 아래로 떨어져 빈 공간을 채우는 부분
for _ in range(m):
for i in range(m-1):
for j in range(n):
if b[i+1][j] == '0': #아래가 비어있다면
b[i+1][j] = b[i][j]
b[i][j] = '0'
return answer
Tip!
(1) string으로 입력 받은 걸 2차원 list로 바꾸기 : 한 요소씩 접근하지 말고 한줄 씩 list() 쓰기
# board를 2차원 list로
for i in range(m):
board[i] = list(board[i])b = [list(x) for x in board]
(2) a에는 b값을, b에는 c값을 줄 때 -> 한 줄로도 표현 가능하다
a = 1
b = 2
a, b=b, 3
print(a) # 출력값: 2
print(b) # 출력값: 3
다른 풀이
아이디어: 전치행렬(행,열이 아니라 열,행으로 읽을 수 있게 회전)
def solution(m, n, board):
board = [[j for j in i] for i in board]
answer = 0 # 몇 개의 블록이 사라졌는 지
switch = 1
while switch:
switch = 0
remove_list = set() #터트릴 블록 저장
for i in range(m - 1):
for j in range(n - 1):
now_box = {
board[i][j],
board[i + 1][j + 1],
board[i + 1][j],
board[i][j + 1],
} # set임
if board[i][j] != -1 and len(now_box) == 1:
# board[i][j]가 -1이면 이미 제거 된거임
switch = 1
remove_list.add((i, j))
remove_list.add((i + 1, j + 1))
remove_list.add((i, j + 1))
remove_list.add((i + 1, j))
answer += len(remove_list)
for i, j in remove_list:
board[i][j] = -1
board = list(zip(*board)) # 배열 회전
board = [[j for j in i if j != -1] for i in board] # -1인 블록 제거하기
board = [[-1] * (m - len(i)) + i for i in board] # 나머지 채우고
board = list(map(list, zip(*board))) #원래대로
return answer
Tip!
(1) 전치행렬 구현
board = [[1,2,3],[4,5,6],[7,8,9]]
board = list(zip(*board))
print(board)
#출력값: [(1, 4, 7), (2, 5, 8), (3, 6, 9)]
(2) -1인 블록 제거하고 채우기
m, n = 4,3
board = [[1,2,3],[-1,5,-1],[7,8,9],[10,11,12]]
board = list(zip(*board))
board2 = [[j for j in i if j != -1] for i in board] # -1인 블록 제거하기
board3 = [[-1] * (m - len(i)) + i for i in board2] # 나머지 채우고
board = list(map(list, zip(*board3))) #원래대로
print(board2) # [[1, 7, 10], [2, 5, 8, 11], [3, 9, 12]]
print(board3) # [[-1, 1, 7, 10], [2, 5, 8, 11], [-1, 3, 9, 12]]
print(board) # [[-1, 2, -1], [1, 5, 3], [7, 8, 9], [10, 11, 12]]
