코드는 Swift로 작성하였습니다.
제가 직접 작성한 코드라 최적의 코드가 아닐 수 있음을 알려드립니다. 🐹
문제는 저작권 관련 문제가 있을 수 있어서 적어두지 않았습니다.
import Foundation
func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
var result = 0
var board = board
var store = [0]
var storeLast = 0
for i in moves {
var index = 0
while true {
if index == board.count { break }
if board[index][i-1] != 0 {
store.append(board[index][i-1])
if storeLast == store.last {
store.popLast()
store.popLast()
result += 2
storeLast = store.last!
}
storeLast = store.last!
board[index][i-1] = 0
break
}
index += 1
}
}
return result
}