본문 바로가기

# 02/Swift - CTP

[Swift] 코딩테스트 연습! Lv2. [1차] 프렌즈4블록

반응형
/* [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가 사용된다.
 출력 형식
 입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.
*/
func solution8(_ m:Int, _ n:Int, _ board:[String]) -> Int {
    var b:[[Character]] = board.map{ Array($0) }
    var result:Int = 0
    while true {
        var list:Set<[Int]> = Set<[Int]>()
        for i in 0...b.count-2 {
            for j in 0...b[i].count-2 {
                if b[i][j] != Character("0") && b[i][j] == b[i][j+1] && b[i][j] == b[i+1][j] && b[i][j] == b[i+1][j+1] {
                    list.insert([i,j])
                    list.insert([i,j+1])
                    list.insert([i+1,j])
                    list.insert([i+1,j+1])
                }
            }
        }
        if list.count == 0 {
            break
        }
        result += list.count
        var sorted = list.sorted{
            if $0[0] < $1[0] {
                return true
            } else if $0[0] == $1[0] {
                if $0[1] < $1[1] {
                    return true
                } else {
                    return false
                }
            } else {
                return false
            }
        }
        for value in sorted {
            var upValue:[Character] = []
            if value[0] > 0 {
                for i in 0...value[0]-1 {
                    if b[i][value[1]] != Character("0") {
                        upValue.append(b[i][value[1]])
                        b[i][value[1]] = "0"
                    }
                }
                if upValue.count == 0 {
                    b[value[0]][value[1]] = "0"
                } else {
                    for i in 0...upValue.count-1 {
                        b[value[0]-i][value[1]] = upValue[upValue.count-1-i]
                    }
                }
            } else {
                b[value[0]][value[1]] = "0"
            }
        }
    }
    return result
}
반응형