[programmers]level 1-クレーン兄抽出ゲーム(lang:swift)


クレーン式油抜きゲーム
問題の説明
ゲーム開発者「Jordy」はクレーン型の抽出機をモバイルゲームに変えようとしている.
ゲームの面白さを高めるために、「Jordie」は画面レイアウトとルールをゲームロジックに反映し、以下のようにします.

ゲーム画面は「1 x 1」サイズのセルからなる「NxN」サイズの正斜角メッシュで、上部にクレーン、右側にバスケットがあります.(上の図は、サイズが「5 x 5」の例を示しています).各格子間には様々なぬいぐるみがあり、ぬいぐるみのない格子は空いています.すべての人形は「1 x 1」の大きさのメッシュを占め、メッシュの一番下から順番に積み上げられます.ゲームユーザーはクレーンを左右に動かし、停止した位置から一番上の人形を持ち上げることができる.拾った人形はかごの中に積み上げられ、かごの一番下の格子から人形は順番にかごの中に積み上げられます.下の図は[1番、5番、3番の位置で人形を順番にかごに入れる様子を示しています.

同じ形の2人の人形が連続してかごに積まれると、2人の人形が爆発してかごから消えてしまいます.上の状態で、次に[5番]の位置で人形をかごに積み上げると、同じ形の人形が2つ消えてしまいます.

クレーンが作動すると、人形が挟まないことはありませんが、人形がないところでクレーンを起動すれば、何も起こりません.また、バスケットが十分大きく、すべての人形を収容できると仮定します.(図中、スクリーン表示制限は5コマのみ表示)
ゲーム画面上の格子状態の2次元配列板と人形を挟むために、起動クレーンの位置を含む配列動作をパラメータとして指定した場合、solution関数を完了し、クレーンをすべて起動させ、爆発して消えた人形の個数を返します.
[制限]
board配列は2次元配列であり、サイズは「5 x 5」または「30 x 30」より大きい.
boardの各セルには、0または100より大きい整数が含まれています.
0はスペースです.
1から100の数字はそれぞれ異なる人形の形を表し、同じ数字は同じ形の人形を表す.
moves配列のサイズは1000を超えない.
moves配列の各要素の値は1より大きく、board配列の水平サイズよりも自然数です.
I/O例
board	moves	result
[[0,0,0,0,0],[0,0,1,0,3],[0,2,5,0,1],[4,2,4,4,2],[3,5,1,3,1]]	[1,5,3,5,1,2,1,4]	4
I/O例説明
I/O例#1
人形の初期状態は、与えられた問題の例と同じである.クレーンは[1,5,3,5,1,2,1,4]号の位置から順に人形を持ち上げ、かごに入れたが、状態は下図のようにかごの中で爆発して消えた人形が4個あった.

===================================
私が書いたコード
import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    
    var store:[Int]  = []
    var count: Int = 0
    var index: Int = 0
    var tempBoard: [[Int]] = board
    
    for move in moves {
        index = 0
        for boardCol in tempBoard {
            if boardCol[move-1] != 0 {
                if (store.last ?? 0) == boardCol[move-1] {
                    store.removeLast()
                    tempBoard[index][move-1] = 0
                    count += 2
                } else {
                    store.append(boardCol[move-1])
                    tempBoard[index][move-1] = 0
                }
                break
            }
            index += 1
        }
    }
    return count
}