我正在尝试编写一个函数,它将一个字符数组(称为单词)和一个字符作为参数,并返回一个“单词”数组,并在所有位置插入字符:
示例
typealias Word = [Character]
func insert(letter: Character, into word: Word) -> [Word] {
}
所以打电话:
insert("c", into: ["a", "b"])
将返回:
[["c", "a", "b"], ["a", "c", "b"], ["a", "b", "c"]]
到目前为止,我提出的最简单的是:
func insert(letter: Character, into word: Word) -> [Word] {
return Array<Int>(0...word.count).map() {
var newWord = word
newWord.insert(letter, atIndex: $0)
return newWord
}
}
我从中获得了通用版本:
extension Array
{
func insert(element: T) -> [[T]] {
return (0...self.count).map() {
var newArray = self
newArray.insert(element, atIndex: $0)
return newArray
}
}
}
然而,这涉及制作输入词的可变副本。它有更好/更优雅/更简单(功能)的方式来实现这个目标吗?
答案 0 :(得分:2)
我会将“非变异数组插入”移动到单独的方法中:
extension Array
{
func arrayByInserting(element: T, atIndex: Int) -> [T] {
var result = self
result.insert(element, atIndex: atIndex)
return result
}
func insert(element: T) -> [[T]] {
return (0 ... self.count).map() { self.arrayByInserting(element, atIndex: $0) }
}
}
然后你的问题(据我所知),如果有更好的方法
arrayByInserting
方法。实际上你的方法对我来说很好。你需要
返回一个新数组,所以你也可以从自我的副本开始。
使用切片可以稍微写更短:
func arrayByInserting(element: T, atIndex: Int) -> [T] {
return self[0 ..< atIndex] + [element] + self[atIndex ..< self.count]
}
然而,快速测试表明,这种方法比你的方法慢10倍 。
答案 1 :(得分:1)
你可以在Swift中找到非常有趣的功能示例: objc.io Functional Snippets
&#34;排列&#34;片段是你正在寻找的。 p>
这是一个简化的例子。但我强烈推荐objc.io片段。
数组扩展:
extension Array {
var decompose : (head: T, tail: [T])? {
return (count > 0) ? (self[0], Array(self[1..<count])) : nil
}
}
这里是通用功能:
func between<T>(x: T, ys: [T]) -> [[T]] {
if let (head, tail) = ys.decompose {
return [[x] + ys] + between(x, tail).map { [head] + $0 }
} else {
return [[x]]
}
}
这可以解决您的问题:
let char: Character = "c"
let charArray = ["a", "b"] as [Character]
let r = between(char, charArray)