Обработайте определения типов по шаблону - Какова Ваша работа вокруг?

Один подход к выравниванию рекурсивной структуры класса с рекурсивной функцией.

Вот класс, который мы хотели бы сгладить:

public class Nested {
    public let n : Int
    public let sub : [Nested]?
    public init(_ n:Int, _ sub:[Nested]?) {
        self.n = n
        self.sub = sub
    }
}

Вот функция, которая демонстрирует как это можно сделать:

func test() {
    let h = [
        Nested(1, [Nested(2, nil), Nested(3, nil)])
    ,   Nested(4, nil)
    ,   Nested(5, [Nested(6, nil), Nested(7, [Nested(8, nil), Nested(9, nil)])])
    ]
    func recursiveFlat(next:Nested) -> [Nested] {
        var res = [Nested]()
        res.append(next)
        if let subArray = next.sub {
            res.appendContentsOf(subArray.flatMap({ (item) -> [Nested] in
                recursiveFlat(item)
            }))
        }
        return res
    }
    for item in h.flatMap(recursiveFlat) {
        print(item.n)
    }
}

Сердцем этого подхода является локальная функция recursiveFlat. Он добавляет содержимое вложенного объекта к результату, а затем условно вызывает себя для каждого элемента, чтобы добавить его содержимое.