如何通过过滤在 swift 中就地改变整数数组

JKR*_*KRT 5 arrays filtering immutability swift

可以快速过滤这样的数组:

var numbers = Array(1...1000000)
numbers = numbers.filter( { return $0 % 2 == 0  } ) 
Run Code Online (Sandbox Code Playgroud)

是否可以过滤并避免在过滤完成时发生的复制操作,例如改变原始数组。

与此伪代码类似: numbers.MutablefilterOperation({ return $0 % 2 == 0})

在 C++ 中,相当于上面 Swift 中发生的事情是:

std::vector<int> originalNumbers(1000000);
std::vector<int> newNumbers;
std::copy_if (originalNumbers.begin(), originalNumbers.end(), std::back_inserter(newNumbers), [](int i) { return i % 2 == 0 } );
Run Code Online (Sandbox Code Playgroud)

出于性能原因,我想实现的目标:

std::vector<int> originalNumbers(1000000);
auto pos = std::remove_if(originalNumbers.begin(), originalNumbers.end(), [](int x) { return x % 2 == 0; });
originalNumbers.erase(pos, originalNumbers.end());
Run Code Online (Sandbox Code Playgroud)

Vad*_*gin 3

此实现应该执行过滤,而不必在此过程中制作整个数组的临时副本(除非它的副本被另一个变量引用,请参阅“写入时复制”)

extension Array {
    mutating func filterInPlace(isIncluded: (Element) throws -> Bool) rethrows {
        var writeIndex = self.startIndex
        for readIndex in self.indices {
            let element = self[readIndex]
            let include = try isIncluded(element)
            if include {
                if writeIndex != readIndex {
                    self[writeIndex] = element
                }
                writeIndex = self.index(after: writeIndex)
            }
        }
        self.removeLast(self.distance(from: writeIndex, to: self.endIndex))
    }
}

// example:
var arr = [6,2,6,5,2,5,6,2,2,1,6,7,3]
arr.filterInPlace { $0 % 2 == 1 }
print(arr) // [5, 5, 1, 7, 3]
Run Code Online (Sandbox Code Playgroud)