Programmers / [1차] 프렌즈4블록 / Python 파이썬
*문제 출처는 프로그래머스에 있습니다.
문제 제목: [1차] 프렌즈4블록
문제 사이트: https://school.programmers.co.kr/learn/courses/30/lessons/17679
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제 설명
프렌즈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,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개의 블록이 지워진다.
# 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임
# 블록이 다 지워질 때 까지 반복
# 영역을 돌면서 같은 것이 네모에 있는지 확인
# 지워지면 위의 블록이 내려옴 (1행~마지막행 까지 돌면서 내려줌)
# 블록이 내려와 생긴 빈 칸은 '0'으로 둔다 -> 내가 정함
# 지워진 블록의 수 리턴
문자열을 리스트 형태로 바꿔준다
new_board = []
for i in board:
arr = list(map(str,i))
new_board.append(arr)
영역을 돌며 같은 문자가 있는지 확인
# 영역을 돌면서 같은 것이 네모에 있는지 확인
row = 1
col = 1
save_arr = [] # 같은 블럭이 있는지 체크
while True:
if row >= (n - 1):
break
if new_board[row - 1][col - 1] == new_board[row - 1][col] == new_board[row][col - 1] == new_board[row][col]:
save_arr.append([row, col])
col += 1
if col > (m - 1):
col = 1
row += 1영역을 돌면서 같은 것이 있는지 체크 후 배열에 저장함
(바꾸면 겹치는 영역을 확인 못하므로 체크만)
체크 후 같은 것이 있던 원소는 0으로 처리
for i in save_arr:
b_row = int(i[0])
b_col = int(i[1])
new_board[b_row - 1][b_col - 1] = "0"
new_board[b_row - 1][b_col] = "0"
new_board[b_row][b_col - 1] = "0"
new_board[b_row][b_col] = "0"
지워진 것은 사라짐 (빈 공간에 위에 있던 원소가 내려옴)
tmp 행렬로 각 행의 원소들을 1차원 리스트로 받아줘서 차례대로 내려준다
# 지워지면 위의 블록이 내려옴 (1행~마지막행 까지 돌면서 내려줌)
for i in range(n):
cnt = m-1
tmp = ["0" for _ in range(m)]
for j in range(m-1,-1,-1):
if new_board[j][i] != "0":
tmp[cnt] = new_board[j][i]
cnt -= 1그리고 열의 형태 (1차원)으로 받았던 리스트를 다시 각 행에 붙여 넣어준다
for k in range(m):
new_board[k][i] = tmp[k]
해당 코드
for i in range(n):
cnt = m-1
tmp = ["0" for _ in range(m)]
for j in range(m-1,-1,-1):
if new_board[j][i] != "0":
tmp[cnt] = new_board[j][i]
cnt -= 1
for k in range(m):
new_board[k][i] = tmp[k]
이러한 과정을 반복하면 됨..
반복할 과정이 없으면 0의 개수(지워진 블록의 수를 리턴하면 됨)
if new_board != board:
solution(m,n,new_board)
else:
for i in new_board:
for j in i:
if j == "0":
answer += 1
return answer
나의 풀이
def solution(m, n, board):
answer = 0
new_board = []
for i in board:
arr = list(map(str, i))
new_board.append(arr)
row = 1 # m
col = 1 # n
save_arr = [] # 같은 블럭이 있는지 체크
while True:
if row == (n - 1):
break
if new_board[row - 1][col - 1] == new_board[row - 1][col] == new_board[row][col - 1] == new_board[row][col]:
save_arr.append([row, col])
col += 1
if col > (m - 1):
col = 1
row += 1
for i in save_arr:
b_row = int(i[0])
b_col = int(i[1])
new_board[b_row - 1][b_col - 1] = "0"
new_board[b_row - 1][b_col] = "0"
new_board[b_row][b_col - 1] = "0"
new_board[b_row][b_col] = "0"
# 지워지면 위의 블록이 내려옴 (1행~마지막행 까지 돌면서 내려줌)
for i in range(n):
cnt = m - 1
tmp = ["0" for _ in range(m)]
for j in range(m - 1, -1, -1):
if new_board[j][i] != "0":
tmp[cnt] = new_board[j][i]
cnt -= 1
for k in range(m):
new_board[k][i] = tmp[k]
if new_board != board:
solution(m, n, new_board)
for i in new_board:
for j in i:
if j == "0":
answer += 1
print(answer)
return answer
if __name__ == "__main__":
m,n,board = 4, 5, ["CCBDE", "AAADE", "AAABF", "CCBBF"]
print(solution(m,n,board))이 코드대로 실행하면 출력 값이
이렇게 뜬다..
다 왔는데 왜 재귀를 활용 못하니 ㅠㅠㅠㅠ
재귀만 제대로 활용하면 풀 거 같은데 잘 안된다...
def solution(m, n, board):
answer = 0
board = [list(i) for i in board]
if not check(m,n,board):
return 0
new_board = set_board(m, n, board)
while check(m,n,new_board):
new_board = set_board(m, n,new_board)
for i in new_board:
for j in i:
if j =="0":
answer += 1
return answer
def set_board(m,n,board):
new_board = []
for i in board:
arr = list(map(str, i))
new_board.append(arr)
row = 1 # m
col = 1 # n
save_arr = [] # 같은 블럭이 있는지 체크
while True:
if row == (n - 1):
break
if new_board[row - 1][col - 1] == new_board[row - 1][col] == new_board[row][col - 1] == new_board[row][col]:
save_arr.append([row, col])
col += 1
if col > (m - 1):
col = 1
row += 1
for i in save_arr:
b_row = int(i[0])
b_col = int(i[1])
new_board[b_row - 1][b_col - 1] = "0"
new_board[b_row - 1][b_col] = "0"
new_board[b_row][b_col - 1] = "0"
new_board[b_row][b_col] = "0"
# 지워지면 위의 블록이 내려옴 (1행~마지막행 까지 돌면서 내려줌)
for i in range(n):
cnt = m - 1
tmp = ["0" for _ in range(m)]
for j in range(m - 1, -1, -1):
if new_board[j][i] != "0":
tmp[cnt] = new_board[j][i]
cnt -= 1
for k in range(m):
new_board[k][i] = tmp[k]
return new_board
def check(m,n,board):
save_arr = [] # 같은 블럭이 있는지 체크
row = 1 # m
col = 1 # n
while row != (n - 1):
if board[row - 1][col - 1] == board[row - 1][col] == board[row][col - 1] == board[row][col]:
if board[row][col] != "0":
save_arr.append([row, col])
col += 1
if col > (m - 1):
col = 1
row += 1
if save_arr:
return True
else:
return False겨우 뜯어고쳤는데 이렇게 된다
다른 분들 풀이를 참고했는데
모든 과정을 함수로 뽀개서 간략하게 끝냈다.
나처럼 무식하게 하나로 한게 아님 ㅠㅠ
알고리즘을 짜는 능력은 어느정도 는 것 같은데 좋은 코드를 짜는 연습을 더 해야할 거 같다.
모범답안(가장 도움이 많이 된 답)
def return_xy(m,n,board): # 2*2 형태로 사라질 블록의 왼쪽위 블럭의 좌표값 리턴
arr = []
for i in range(m):
for j in range(n):
if i+1 < m and j+1 < n:
if board[i][j] and board[i][j] == board[i][j+1] == board[i+1][j] == board[i+1][j+1]:
# 왼쪽위 블록 기준으로 그 값이 0이 아니고 오른쪽, 아래, 오른쪽 아래 블록의 값이 값으면 좌표값 저장
arr.append([i,j])
return arr
def check(m,n, board): # 2*2 형태로 사라질 블록이 있는지 확인
for i in range(m):
for j in range(n):
if i+1 < m and j+1 < n:
if board[i][j] and board[i][j] == board[i][j+1] == board[i+1][j] == board[i+1][j+1]:
return True
return False
def compression(arr): # 2*2블럭의 값이 0(빈공간)으로 채워지고 그 위의 블럭이 빈공간을 매꿈
# Transpose 된 배열을 처리
# 예를 들어, ['T',0,0,'T','T','T'] 인 열을 [0,0, 'T','T','T','T'] 형태로 만들어줌
new_arr = []
for i in arr:
cnt = i.count(0)
if cnt == 0:
new_arr.append(i)
continue
temp = []
for j in i:
if j != 0:
temp.append(j)
temp = [0]*cnt + temp
new_arr.append(temp)
return new_arr
def Transpose(arr): # compression 함수를 쉽게 하기 위해 행과 열을 바꿔줌
A = list(zip(*arr)) # Transpose 시키는 방법
new_arr = [] # Tuple은 값을 바꿀 수 없어 다시 List 형태로 저장해줌
for i in A:
new_arr.append(list(i))
return new_arr
def solution(m, n, board):
# 기존의 board 형태는 문자열 배열로 값을 변경 할 수 없으므로 character 배열로 바꿔줌
# 예) ["TA", "TT"] -> [['T', 'A'], ['T', 'T']]
board = [list(i) for i in board]
if not check(m,n, board): # 2*2 형태로 지워질 블록이 없다면 return 0
return 0
while check(m,n, board): # 2*2 형태로 지워질 블록이 있다면
arr = return_xy(m,n, board) # 사라져야 할 블록의 왼쪽위 블록의 좌표값을 얻어옴
for x,y in arr: # 지워질 블록의 값을 0으로 바꿔줌
board[x][y] = 0
board[x][y+1] = 0
board[x+1][y] = 0
board[x+1][y+1] = 0
board_T = Transpose(board) # 압축을 쉽게하기 위해 행과 열을 바꿔줌
board_T = compression(board_T) # 압축
board = Transpose(board_T) # 다시 원상태로 바꿔줌
# board에 지워진 블록이 몇개인지 count 함
cnt = sum([board[i].count(0) for i in range(len(board))])
# 답 return
return cnt