[프로그래머스][LV.2] [1차] 프렌즈4블록 | python3

문제링크:  [1차] 프렌즈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가 사용된다.

 

 

---

 

문제풀이

def solution(m, n, board):
    
    nm=[]
    for i in range(m-1):
        for j in range(n-1):
            if board[i][j]==board[i+1][j]==board[i][j+1]==board[i+1][j+1]:
                nm.append((i,j))
                nm.append((i+1,j))
                nm.append((i,j+1))
                nm.append((i+1,j+1))
                
    answer=len(set(nm))
    return answer

 

처음 블록에서 옆 아래 대각선이 같으면 리스트에 좌표를 추가하여 중복을 걸러내는 식으로 하려고 했었다.

근데 블록이 없어지고 떨어진 후에도 같은게 4개 있으면 없어지는 듯....

 

없앤다음에 밑으로 내려줘야함. while과 조건문(인덱스)를 활용하여 풀이.

나의코드

def solution(m, n, board):
    
    # 문자열까지 리스트로 변환. 이중배열
    for i in range(m):
        board[i]=list(board[i])
        
    cnt=0
    rm=set()
    while True:
        for i in range(m-1):
            for j in range(n-1):
                t=board[i][j]
                if not t:
                    continue
                if board[i+1][j]==board[i][j+1]==board[i+1][j+1]==t:
                    rm.add((i,j))
                    rm.add((i+1,j))
                    rm.add((i,j+1))
                    rm.add((i+1,j+1))
        if rm:
            cnt+=len(rm)
            for i,j in rm:
                board[i][j]=[]
            # rm 초기화
            rm=set()
        #같은 4개가 없을때
        else:
            return cnt
        
        while True:
            move=0
            for i in range(m-1):
                for j in range(n):
                    if board[i][j] and board[i+1][j]==[]:
                        board[i+1][j]=board[i][j]
                        board[i][j]=[]
                        move=1
            if move==0:
                break

 

---

 

참고하기

def solution(m, n, board):
    board.reverse()
    field = [list(cols) for cols in zip(*board)]
    answer = 0

    while True:
        bomb = set()
        for row in range(n - 1):
            for col in range(m - 1):
                try:
                    if field[row][col] == field[row + 1][col] == field[row][col + 1] == field[row + 1][col + 1]:
                        bomb.update({(row, col), (row + 1, col), (row, col + 1), (row + 1, col + 1)})
                except:
                    break

        if not len(bomb):
            break

        for r, c in bomb:
            field[r][c] = ''
            answer += 1

        for row in range(n):
            field[row] = list(''.join(field[row]))

    return answer

def pop_num(b, m, n):
    pop_set = set()
    # search
    for i in range(1, n):
        for j in range(1, m):
            if b[i][j] == b[i - 1][j - 1] == b[i - 1][j] == b[i][j - 1] != '_':
                pop_set |= set([(i, j), (i - 1, j - 1), (i - 1, j), (i, j - 1)])
    # set_board
    for i, j in pop_set:
        b[i][j] = 0
    for i, row in enumerate(b):
        empty = ['_'] * row.count(0)
        b[i] = empty + [block for block in row if block != 0]
    return len(pop_set)


def solution(m, n, board):
    count = 0
    b = list(map(list, zip(*board)))  # 행열 바꿔치기
    while True:
        pop = pop_num(b, m, n)
        if pop == 0: return count
        count += pop