使我的数组为零，并保持一些索引不变

Question

我有一个数组 'board_matrix'，里面有一些值。我还有另一个数组 'win' 和一些索引值。现在我想让 'board_matrix' 全部为零，除了 'win' 数组中的那些 索引 。

说，

board_matrix = [1,0,2,2,1,0,1,0,1]
win = [0,4,8]

then output should be 
new_array = [1,0,0,0,1,0,0,0,1]

Answer 1

如果在 win 数组中找不到索引，您可以遍历 board_matrix 数组并覆盖值。像这样：

for (i=0; i<board_matrix.size; i++) {
    if (!win.contains(i)) board_matrix[i] = 0;
}

Answer 2

board_matrix.enumerated().map { [winIndices = Set(win)] in
  winIndices.contains([=10=].offset) ? [=10=].element : 0
}