使用无序的整数数组,如下所示:
let numbers = [4, 3, 1, 5, 2]
是否可以在 Swift 中使用 .sort { } 对数组进行排序,其中一项优先并放置在数组的第一个索引中。所以我们可以得到 [3, 1, 2, 2, 4, 5],而不是返回 [1, 2, 3, 4, 5]?
你可以声明一个这样的函数:
func sort(_ array: [Int], prioritizing n: Int) -> [Int] {
var copy = array
let pivot = copy.partition { $0 != n }
copy[pivot...].sort()
return copy
}
它使用partition(by:)
功能。
你可以这样使用它:
let numbers = [4, 3, 1, 5, 2]
let specialNumber = 3
sort(numbers, prioritizing: specialNumber) //[3, 1, 2, 4, 5]
以下是一些测试用例:
sort([3, 3, 3], prioritizing: 3) //[3, 3, 3]
sort([9, 4, 1, 5, 2], prioritizing: 3) //[1, 2, 4, 5, 9]
这里有一个仅使用sorted(by:)
的替代解决方案:
let numbers = [4, 3, 1, 5, 2]
let vipNumber = 3
let result = numbers.sorted {
($0 == vipNumber ? Int.min : $0) < ($1 == vipNumber ? Int.min : $1)
}
print(result) //[3, 1, 2, 4, 5]