Swift数组是否有内置方法可拆分为有序的前缀与后缀？

嘿，这个问题问得相当精准！刚好我对Swift数组的这些操作熟得很，来给你好好唠唠～

首先直接给结论：Swift标准库并没有直接提供一个名为cut的内置方法，但我们可以用现有工具组合出完全符合你需求的功能，甚至可以自己扩展一个方法来实现你例子里的调用方式。

先说说你提到的partition和split为什么不适用：

• partition(by:)会根据条件重新排列数组元素，把满足条件的移到前面，但它会打乱原有的元素顺序，这显然不是你想要的；
• split(separator:)或者split(whereSeparator:)是按分隔符拆分出多个子数组，没法直接实现“找到第一个不满足条件的位置，把数组切成前后两部分”的需求，而且拆分后的大小也不固定。

方案1：用现有标准库方法组合实现 #

Swift的prefix(while:)和drop(while:)这两个方法完全能满足你的需求，而且它们都会严格保留原数组的元素顺序：

• prefix(while:)会返回数组中从开头开始连续满足条件的元素；
• drop(while:)会返回数组中去掉开头连续满足条件的元素后剩下的部分。

直接上代码例子：

let numbers = [1,2,3,5,6,2,3,5]
// 获取前缀：所有开头连续小于5的元素
let prefixPart = numbers.prefix(while: { $0 < 5 })
// 获取后缀：去掉开头连续小于5的元素后剩下的部分
let suffixPart = numbers.drop(while: { $0 < 5 })

print((prefixPart, suffixPart))
// 输出结果：([1, 2, 3], [5, 6, 2, 3, 5])

方案2：给Array扩展一个cut方法 #

如果觉得每次写两行代码麻烦，我们可以给Array扩展一个自定义的cut(while:)方法，让它直接返回拆分后的前后两部分，就像你例子里展示的那样：

extension Array {
    func cut(while predicate: (Element) -> Bool) -> (prefix: [Element], suffix: [Element]) {
        // 找到第一个不满足条件的元素的索引
        let splitIndex = firstIndex(where: { !predicate($0) }) ?? endIndex
        // 拆分并返回两个数组
        return (Array(self[..<splitIndex]), Array(self[splitIndex...]))
    }
}

// 调用方式超简洁
let numbers = [1,2,3,5,6,2,3,5]
let result = numbers.cut(while: { $0 < 5 })
print(result.prefix) // [1, 2, 3]
print(result.suffix) // [5, 6, 2, 3, 5]

这个扩展方法的逻辑很清晰：用firstIndex(where:)定位到第一个不满足条件的元素位置，如果所有元素都满足条件，就用endIndex作为拆分点，此时后缀就是空数组，完全符合预期。而且整个过程完全保留了原数组的元素顺序，所有元素都被分到前缀或后缀中，没有丢失。

内容的提问来源于stack exchange，提问作者SwiftsNamesake