프로그래머스_크레인 인형뽑기

hankyulee·2021년 11월 17일
0

Swift coding test 준비

목록 보기
43/57

import Foundation
var result = [Int]()
var original = [Int]()
func boadmodify(board: [[Int]], col : Int)->[[Int]] {
    var newBoard = board
    var tmpResult = result
    var size = newBoard.count
    for row in (0..<size){
        if newBoard[row][col] != 0 {   
            if result.count != 0 {
                if result.last == newBoard[row][col] {
                    original.append(newBoard[row][col])
                    result.popLast()       
                }   
                else {
                    result.append(newBoard[row][col])
                    original.append(newBoard[row][col])
                }
            }
            else{result.append(newBoard[row][col])
                original.append(newBoard[row][col])
                               }
            newBoard[row][col] = 0
            break
        }
    }
    return newBoard
}

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {

    var resultBoard = board
    for move in moves { 
        resultBoard = boadmodify(board: resultBoard, col: move-1)
    }
    return original.count - result.count
}


0개의 댓글