frienfs4block_je.py

def solution(m, n, board):
    answer = 0
    dx = [0, 1, 1]
    dy = [1, 0, 1]
    check = True
    tmp = board
    board = []
    for b in range(len(tmp)):
        board.append(list(tmp[b].strip()))
    findloc = []
    while check:
        findloc.clear()
        for x in range(m-1):
            for y in range(n-1):
                cnt = 0
                if board[x][y] == ' ':
                    continue
                for t in range(3):
                    if board[x][y] == board[x+dx[t]][y+dy[t]]:
                        cnt += 1
                    else:
                        break
                if cnt == 3:
                    findloc.append([x, y])
        if len(findloc):
            for f in range(len(findloc)):
                board[findloc[f][0]][findloc[f][1]] = 'x'
                board[findloc[f][0]+dx[0]][findloc[f][1]+dy[0]] = 'x'
                board[findloc[f][0]+dx[1]][findloc[f][1]+dy[1]] = 'x'
                board[findloc[f][0]+dx[2]][findloc[f][1]+dy[2]] = 'x'
            for x1 in range(m):
                for y1 in range(n):
                    if board[x1][y1] == 'x':
                        answer += 1
                        for x2 in range(x1, -1, -1):
                            if x2 == 0:
                                board[x2][y1] = ' '
                            else:
                                board[x2][y1] = board[x2-1][y1]
                                board[x2 - 1][y1] = ' '
        else:
            check=False
    return answer


if __name__ == '__main__':
    answer = solution(4, 5, ['CCBDE', 'AAADE', 'AAABF', 'CCBBF'])
    print('출력값: ', answer)