Swift - 插入字符数组,功能解决方案

时间:2014-12-14 12:09:49

标签: arrays generics swift functional-programming

我正在尝试编写一个函数,它将一个字符数组(称为单词)和一个字符作为参数,并返回一个“单词”数组,并在所有位置插入字符:

示例

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
        }
    }
}

然而,这涉及制作输入词的可变副本。它有更好/更优雅/更简单(功能)的方式来实现这个目标吗?

2 个答案:

答案 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;片段是你正在寻找的。

这是一个简化的例子。但我强烈推荐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)