arrays - Swift 中的数组操作和重新排序

Question

给定一个数组(例如，[ 1, 0, 2, 0, 0, 3, 4 ])，实现方法 将非零元素移动到数组的开头(其余元素无关紧要)

我已经按如下方式实现，它有效但我想知道更短的方法吗？

import Foundation

var inputArray = [ 1, 0, 2, 0, 0, 3, 4 ]

func remoZeros (inputArray :[Int]) -> [Int]
{
  var nonZeroArray = [Int]()
  var zeroArray = [Int]()

  for item in inputArray
  {
    if item != 0
    {
      nonZeroArray.append(item)
    }
    else
    {
      zeroArray.append(item)
    }
  }

return nonZeroArray + zeroArray

}

var result = remoZeros (inputArray: inputArray)

Answer 1

你可以试试

var inputArray = [ 1, 0, 2, 0, 0, 3, 4 ]

func remoZeros (inputArray :[Int]) -> [Int] {

   return inputArray.filter{$0 != 0} + inputArray.filter{$0 == 0}
}