如何在 Swift 中合并多个数组？

Question

我有几个类是这样的：

A类{} A1 类：A {} A2 类：A {} A3 类：A {} A4 类：A {} 类主{ 变量 a1 : A1 变量 a2 : A2 变量 a3s：[A3] var a4s：[A4] 函数 getAll() -> [A] { 返回 ([a1, a2] + a3s + a4s) } }

如果你看一下函数 getAll()，你会看到我尝试返回一个类型为基类 A 的所有对象的数组。但是，我总是得到错误：

“二元运算符'+(::)'不能应用于'[Any]'和'[a3s]'类型的操作数”

你知道在这种情况下正确的方法是什么吗？

Answer 1

我猜这只是正确转换数组的问题，您可以这样做解决它：

func getAll() -> [A] {
    return ([a1, a2] as [A] + a3s as [A] + a4s)
}

---

只是为了好玩：如果你想动态解决它，你可以这样使用反射：

class A {}
class A1 : A {}
class A2 : A {}
class A3 : A {}
class A4 : A {}

class main {
    var a1 = A1()
    var a2 = A2()
    var a3s : [A3] = [A3()]
    var a4s : [A4] = [A4(), A4()]

    func getAll() -> [A] {
        var res = [A]()
        Mirror(reflecting:self).children.forEach {
            if let a = $0.value as? A {
                res.append(a)
            } else if let aArray = $0.value as? [A] {
                res.append(contentsOf: aArray)
            }
        }
        return res
    }
}

let all = main().getAll()
print(all) //A1, A2, A3, A4, A4

Answer 2

我猜这可能是一种更清洁的方式。

func getAll() -> [A] {
    let all: [[A]] = [[a1], [a2], a3s, a4s]
    return all.flatMap{$0}
}

这会创建一个数组数组，然后使用flatMap 将它们展平为一个数组。

Answer 3

问题在于编译器认为您正在尝试添加[Any] 和[A3]。这是因为[a1, a2] is [Any]和a3s is [A3]的类型。但是运营商+ 做不到。

正如其他人指出的那样，您可以将语句中的数组转换为 [A]，以便 + 运算符可以完成他的工作。

由于您提到您有多个类，因此创建一个可以重用并且可以正确推断类型的函数可能是个好主意。

func merge<T>(_ array: [T]...) -> [T] {
    return array.flatMap { $0 }
}

func getAll() -> [A] {
    return merge([a1, a2], a3s, a4s)
}