Programmers / [1차] 프렌즈4블록 / Python 파이썬

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: [1차] 프렌즈4블록

문제 사이트: https://school.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가 사용된다.

출력 형식

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

입출력 예제

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

---