Swiftで配列(Array)の差分を取得する

Swiftで配列(Array)の差分(diff相当)を取得する

出典はこちらです。
Swiftのバージョンが低くそのまま使えなかったので、すこし手直しして共有しておきます。

extension Array where Element: Hashable {
	typealias E = Element
	func diff(_ other: [E]) -> [E] {
		let all = self + other
		var counter: [E: Int] = [:]
		all.forEach { counter[$0] = (counter[$0] ?? 0) + 1 }
		return all.filter { (counter[$0] ?? 0) == 1 }
	}
}

let a = [1, 2, 3, 4]
let b = [8, 2, 3, 9]
let diff1 = a.diff(b) // => [1, 4, 8, 9]
let diff2 = b.diff(a) // => [8, 9, 1, 4]

Swiftで配列(Array)の差分(引き算)をする

Setのsubtract/subtracting相当のメソッドです。
こちらは新しく作りました。
比較のため、利用する配列はEquatableに準拠している必要があります。

extension Array where Element: Equatable {
	typealias E = Element

	func subtracting(_ other: [E]) -> [E] {
		return self.flatMap { element in
			if (other.filter { $0 == element }).count == 0 {
				return element
			} else {
				return nil
			}
		}
	}

	mutating func subtract(_ other: [E]) {
		self = subtracting(other)
	}
}

var a = [1, 2, 3, 4]
let b = [8, 2, 3, 9]

let sub1 = a.subtracting(b)
sub1 // => [1, 4]
a // => [1, 2, 3, 4]
a.subtract(b)
a // => [1, 4]