为HackerRink解决方案(Swift)重构代码

Question

我正试图解决一个非常简单的HackerRank测试，我有一些代码，我相信是非常干净和重构的...但是它总是在最后一个测试用例中超时。我下载了它的输入，它是一个非常大的值串，但是其他人用更复杂的代码也能很好地通过。如果有的话，我想找一种更简洁的方式来表达我的解决方案。

这就是问题所在：https://www.hackerrank.com/challenges/ctci-array-left-rotation

对size数组的左旋转操作将数组的每个元素单位向左移动。例如，如果对数组执行左旋转，那么数组将变成。

给定一个整数和一个数字数组，对数组执行左旋转。然后将更新后的数组打印为一行以空格分隔的整数。

输入格式

第一行包含两个用空格分隔的整数，分别表示(整数个数)和(必须执行的左旋转次数)的值。第二行包含空格分隔的整数，描述数组初始状态的各个元素。

样本输入

5 4 1 2 3 4 5

抽样输出

5 1 2 3 4

以下是我的解决方案:

import Foundation

func shift(_ a: [String], n: Int, d: Int) -> String {
    var s = a
    for _ in 0 ..< d {
        let v = s.remove(at: 0)
        s.insert(v, at: n-1)
    }
    return s.joined(separator: " ")
}

let values = readLine()!.components(separatedBy: " ").map { Int($0)! }
let line2 = readLine()!.components(separatedBy: " ")
print(shift(line2, n: values[0], d: values[1]))

谁能帮我找到一个更干净的方法来解决这个问题？我的解决方案在最后两个测试用例中的一个总是超时。

先谢谢你。

Answer 1

您不需要进行交换，因为您已经在创建数组（var s）的副本，所以只需使用它。即从源代码中获取并将输出转换为s，漂亮而简单的示例：

func shift(_ a: [String], n: Int, d: Int) -> String {
    var s = a
    for i in 0 ..< n {
        s[i] = a[(i + d)%n]
    }
    return s.joined(separator: " ")
}

Answer 2

根据使用的数据结构，删除/插入应该采用O（n）的元素，而不是在数组上进行交换。

func shift(_ a: [String], n: Int, d: Int) -> String {
    var s = a
    for i in 0 ..< d {
        var t = s[0]
        s[0] = a[(i - d)%n]
        a[(i - d)%n] = t
    }
    return s.joined(separator: " ")
}

我没有运行这个，所以要确保代码是正确的，但这个想法就在那里。