2018 KAKAO BLIND RECRUITMENT
https://programmers.co.kr/learn/courses/30/lessons/17679?language=python3
[문제]
입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라.
[문제 설명]
블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈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개의 블록이 지워진다.
첫 번째 시도 _ 성공
graph 방식이 떠올라서 바로 해결한 문제
하지만, 과연 이 방식이 맞을까 했지만, 데이터가 작아서인지, 생각보다 시간이 오래 걸리지 않아 이 코드로 제출
- 입력받은 board의 줄별로 list형식으로 변환
- 제거했음을 확인할 수 있도록 remove 배열 선언과 함께 초기화
- 제거 후 위에서 블록이 내려오는데, 내려올 블록이 없는 경우 board[i][j] = 0
- if board[i][j] != 0 and board[i][j]가 [i][j+1], [i+1][j], [i+1][j+1] 값과 모두 같은 경우, 블록 제거
- remove 배열의 각 인덱스 해당 값 1로 설정
- if board[i][j] != 0 and board[i][j]가 [i][j+1], [i+1][j], [i+1][j+1] 값과 모두 같은 경우, 블록 제거
- count = 0으로 초기화한 후, 줄별로, remove[i] 값이 1인 것 sum: 제거한 블록 개수 구하기 → answer += count
- 개수 구한 후에는, 제거한 블록 위치에 블록 채우는 과정 수행
- remove[i][j] == 1이면, x = i-1
- x >= 0 and remove[x][j] == 1이면, x -= 1
- x < 0인 경우, board[i][j] = 0
- 아닌 경우, board[i][j] = board[x][j], remove[x][j] = 1
- remove[i][j] == 1이면, x = i-1
- return answer
def solution(m, n, board):
answer = 0
for i in range(len(board)):
board[i] = list(board[i])
while True:
remove = [[0] * n for _ in range(m)]
for i in range(m-1):
for j in range(n-1):
if board[i][j] != 0 and board[i][j] == board[i][j+1] and board[i][j] == board[i+1][j] and board[i][j] == board[i+1][j+1]:
remove[i][j], remove[i][j+1], remove[i+1][j], remove[i+1][j+1] = 1, 1, 1, 1
count = 0
for i in range(m):
count += sum(remove[i])
answer += count
if count == 0:
break
for i in range(m-1, -1, -1):
for j in range(n):
if remove[i][j] == 1:
x = i-1
while x >= 0 and remove[x][j] == 1:
x -= 1
if x < 0:
board[i][j] = 0
else:
board[i][j] = board[x][j]
remove[x][j] = 1
return answer
최종 코드
'Python ◡̈ > 차근차근 Python' 카테고리의 다른 글
[백준 python] 2302 극장좌석 (0) | 2021.08.25 |
---|---|
[programmers python] 17677 뉴스 클러스터링 (0) | 2021.07.22 |
[programmers python] 17680 캐시 (0) | 2021.07.22 |
[programmers python] 42890 후보키 (0) | 2021.07.15 |
[programmers python] 42889 실패율 (0) | 2021.07.15 |