[Swift] 2019 KAKAO BLIND RECRUITMENT 블록 게임

Problem

 

코딩테스트 연습 - 블록 게임

---

Solution

 

1. 먼저 검은 블록을 떨어뜨려 삭제가 가능한 블록들을 찾아낸다.

 

주어진 블록은 차례대로 1번 블록의 0,1,2,3 타입, 2번 블록의 0,1,2,3 타입 , 3번 블록의 0,1,2,3타입이 있다고 가정할 때

 

검은 블록을 위에서 떨어뜨려서 채워야 하는 것이므로 1번 블록의 2,3 타입, 2번 블록의 1,2 타입 3번 블록의 0번 타입만 삭제가 가능합니다.

 

 

2. 주어진 보드에서 블록의 종류, 타입, 세로or가로, 위치를 찾아낸다.

 

아래와 같이 블록은 행과 열을 통해서 세로로 있는지 가로로 있는지를 구별할 수 있습니다.

 

이것을 통해 먼저 가로로 된 블록들을 전체 보드에서 찾습니다.

 

 

 

먼저 세로로 된 블록을 찾아내는 방법은 아래와 같이 2x3 모양을 (0,0) 부터 (N-1,N-1) 까지 확인해서 똑같은 숫자가 4개가 있다면 블록이 존재한다는 의미입니다.

 

 

가로로 된 블록은 3x2 모양을 세로와 마찬가지로 (0,0) 부터 (N-1,N-1) 까지 확인해서 찾아줍니다.

 

 

* 블록을 찾는 함수

 

func findBlocks(_ board:[[Int]],_ isVertical:Bool) {
    let row = isVertical ? 3 : 2
    let col = isVertical ? 2 : 3
    
    for i in 0...board.count-row {
        for j in 0...board.count-col {
            var block:[[Int]] = Array(repeating: [], count: row)
            for k in 0..<row {
                block[row-1-k] = Array(board[i+k][j...j+col-1])
            }
            let change = isExistEqual4Number(block)
            if let change = change {
                let equal = checkEqualBlock(change, isVertical: isVertical, position: CGPoint(x: j, y:i+row-1))
                if let equal = equal {
                    blocks.append(equal)
                }
            }
        }
    }
}

 

*  똑같은 숫자가 4개 있는지 판별하는 함수

 

func isExistEqual4Number(_ block:[[Int]]) -> [[Int]]? {
    var numberCount:[Int:Int] = [:]
    var changeBlock:[[Int]] = []
    for b in block {
        for n in b {
            if numberCount[n] == nil && n != 0 {
                numberCount[n] = 1
            }else {
                numberCount[n]! += 1
            }
        }
    }
    
    for (n,c) in numberCount {
        if c == 4 {
            for (i,b) in block.enumerated() {
                changeBlock.append([])
                for e in b {
                    let oneOrZero = e == n ? 1 : 0
                    changeBlock[i].append(oneOrZero)
                }
            }
            return changeBlock
        }
    }
    return nil
}

 

* 어느 종류의 블록인지 확인하고 해당 블록에 맞게 바꿔주는 함수

 

func checkEqualBlock(_ block:[[Int]],isVertical:Bool,position:CGPoint) -> Block? {
    let type = isVertical ? [1,3] : [0,2]
    for i in 1...3 {
        for t in type {
            if numberTypes[i]![t] == block {
                return Block(position: position, number: i, type: t, isVertical:isVertical)
            }
        }
    }
    return nil
}

 

3. 찾아낸 블록 중 삭제 가능한지 여부를 판단한다.

 

블록이 삭제가 가능한지를 판별하기 위해선 우선 1에서 구한 1번 블록의 2,3 타입, 2번 블록의 1,2 타입 3번 블록의 0번 타입 중 하나여야 하고, 해당 블록의 빈칸의 위에 아무런 블록이 없어야 합니다.

 

* 총 5개 타입의 블록의 위 칸에 아무런 블록이 없는지 확인하는 함수

 

func checkAboveBlocks(_ block:Block,_ board:inout [[Int]]) {
    var xs:[Int] = []
    let x = Int(block.position.x)
    let y = Int(block.position.y)
    var remove:[(Int,Int)] = []
    if block.number == 1 && block.type == 2{
        xs = [x+1,x+2]
        remove = [(x,y),(x+1,y),(x+2,y),(x,y-1)]
    }else if block.number == 1 && block.type == 3 {
        xs = [x]
        remove = [(x,y),(x+1,y),(x+1,y-1),(x+1,y-2)]
    }else if block.number == 2 && block.type == 1 {
        xs = [x+1]
        remove = [(x,y),(x+1,y),(x,y-1),(x,y-2)]
    }else if block.number == 2 && block.type == 2 {
        xs = [x,x+1]
        remove = [(x,y),(x+1,y),(x+2,y),(x+2,y-1)]
    }else if block.number == 3 && block.type == 0 {
        xs = [x,x+2]
        remove = [(x,y),(x+1,y),(x+2,y),(x+1,y-1)]
    }else {
        return
    }
    for x in xs {
        for y in 0...y-1 {
            if board[y][Int(x)] != 0 {
                return
            }
        }
    }
    for r in remove {
        board[r.1][r.0] = 0
    }
    count += 1
}

 

4. 삭제 가능한 블록의 갯수를 반환한다.

 

찾아낸 블록 전체에서 삭제 가능한 블록의 갯수를 찾아서 반환합니다.

 

* 삭제 가능한 블록의 갯수를 구하는 함수

 

func countRemovableBlocks(_ board:[[Int]]) -> Int {
    var board = board
    let sorted = blocks.sorted {$0.position.y < $1.position.y}
    for block in sorted {
        checkAboveBlocks(block,&board)
    }
    return count
}

---

Source Code