[프로그래머스 level2 프렌즈4블록] filter와 lambda를 활용해보자!

오늘 프렌즈4블록과 뒤에 이어지는 문제인 캐시를 풀었다.

캐시보다는 좀 더 많은 아이디어를 요구했던 것 같다❗️

 

오늘 해결해야하는 문제는 다음과 같았다.

programmers.co.kr/learn/courses/30/lessons/17679

코딩테스트 연습 - [1차] 프렌즈4블록

 

 

[문제 설명 ↓]

프렌즈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

 

 

문제를 읽고 생각을 많이할 수 밖에 없었다..

 

2x2의 블록이 같을 경우 위에서 아래로 블록이 떨어지는 형식이기 때문에 행 단위가 아닌 열 단위로 검사해주기 위해 행과 열을 뒤집어 temp에 저장하였다.

이 때 주의할 점이 위에서 아래로 떨어지므로 나중에 pop되고 블록을 비교할 때 칸 수를 쉽게 맞추어 판별하기 위하여 reverse시켰다.

 

 

mark는 pop될 칸들을 표시해두는 배열로 0으로 초기화시켰지만 pop 될 칸은 미래에 1로 칠해준다.

flag는 pop될 칸이 없다면 0이 유지되고 pop될 칸이 있다면 1로 하여,  모든 판을 확인하였는데 pop될 것이 없다면 break 할 것이다.

 

2x2칸을 비교하므로, i와 i+1을 비교하므로, i는 len(temp)-1보다 작아야한다.

[i]에서의 j, j+1 [i+1]에서의 j,j+1을 비교하므로 j는 len(temp[i])-1보다 작아야하고, len(temp[i+1])-1보다 작아야 한다.

2x2칸을 비교하여 모두 똑같은 문자가 있다면 mark에서 해당 위치를 1로 칠해주고, pop되는 것이 있다는 표시인 flag를 1로 해주었다.

 

모든 수정이 끝나면 mark에서 1의 개수가 Pop이 되는 문자 수이므로 answer를 더해주었다.

 

 

pop된 문자가 있는 경우,

temp를 새로 업데이트해주었다.

temp는 pop되야 하는 위치를 '0'으로 임시로 채워두고, filter와 lambda를 활용하여 해당 칸을 한 번에 pop해주었다. (지워줬다.)

--> 이를 위하여 나는 행과 열을 바꾸었다..!

temp[i]=list(filter(lambda x : x !='0',  temp[i]))

살짝 뜯어보면 temp[i] 줄을 '0'이 있는 칸을 제외하고 남겨서 list로 다시 temp[i]를 업데이트 시킨다는 의미이다.

 

 

 

아이디어를 바탕으로 최종 구현한 코드는 다음과 같다.

def solution(m, n, board):
    answer = 0
    temp = [[0 for a in range(m)] for b in range(n)]
    board.reverse()
    for i in range(n):
        for j in range(m):
            temp[i][j] = board[j][i]

    while True :
        mark =  [[0 for a in range(m)] for b in range(n)]
        flag=0
        for i in range(len(temp)):
            for j in range(len(temp[i])):
                if i < len(temp)-1 and j < len(temp[i])-1 and j<len(temp[i+1])-1:
                    if temp[i][j] == temp[i][j+1] == temp[i+1][j] == temp[i+1][j+1] :
                        mark[i][j] = 1
                        mark[i][j+1] = 1
                        mark[i+1][j]=  1
                        mark[i+1][j+1]=1
                        flag = 1

        for i in range(len(mark)):
            answer += sum(mark[i])

        if flag == 0:
            break

        for i in range(len(temp)):
            for j in range(len(temp[i])):
                if mark[i][j]==1:
                    temp[i][j] = '0'
            temp[i]=list(filter(lambda x : x !='0',  temp[i]))

    return answer

 

요즘 카카오 시험 문제를 끝까지 혼자 힘으로 풀어내는 것이 신기하고, 뭔가 점점 풀 수 있는 능력이 향상되고 있는 것 같아서 기분이 굉장히 좋은 것 같다..ㅎㅎ

꾸준히 하나이상씩 푸는 습관을 이어갈 수 있기를...⭐️