我正在尝试编写一个函数,它将一个字符数组(称为单词)和一个字符作为其参数,并返回一个“单词”数组,并在所有位置插入字符: 例 typealias Word = [Character]func insert(letter: Character, into w
例
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 } } }
但是,这涉及制作输入词的可变副本.它有更好/更优雅/更简单(功能)的方式来实现这一目标吗?
我会将“非变异数组插入”移动到一个单独的方法: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倍.