프로그래머스. 프렌즈 4블록

https://school.programmers.co.kr/learn/courses/30/lessons/17679

프로그래머스

 

전체 코드

한승협님 코드 

def solution(m, n, board):
    cnt = 0
    board = [list(i) for i in board]
    
    while True:
        rem = set()
        
        # 제거할 타일을 rem 에 담는 과정
        for x in range(m-1):
            for y in range(n-1):
                if board[x][y] == board[x+1][y] == board[x][y+1] == board[x+1][y+1]:
                    if board[x][y] != "X":
                        rem.add((x,y))
                        rem.add((x+1,y))
                        rem.add((x,y+1))
                        rem.add((x+1,y+1))
                        
        # X 로 바꿈
        cnt += len(rem)
        for x, y in rem:
            board[x][y] = "X"
        
        # 아래 X 칸이 아닌 X 칸을 아래 칸과 맞바꿈
        while True:
            moved = 0
            for x in range(m-1):
                for y in range(n):
                    if board[x+1][y] == "X" and board[x][y] != "X":
                        board[x+1][y], board[x][y] = board[x][y], board[x+1][y]
                        moved += 1
            # X 칸을 다 내려줬다면 움직임이 없음
            if moved == 0:
                break
                
        # 제거할 칸이 없다면 나감
        if not rem:
            break
        
    return cnt