Swift：将数组 [A、B、B、B、A、B、B、B] 转换为哈希数组 [[A: [B、B、B] ]，[A: [B、B、B]] ]

Question

我有两种结构体，Header和Session，都符合协议TimelineItem。
我的一个Array由TimelineItem组成，像这样：
[Header1，SessionA，SessionB，Header2，SessionC，SessionD]

我需要将Session分组到相关的Header下，就像这样：
[[Header1：[SessionA，SessionB]，[Header2：[SessionC，SessionD]]]

我尝试使用filter方法检索仅Header结构，并使用split方法检索Session数组的数组。 这些方法都可以正常工作，但我无法管理如何将它们协调起来构建我的最终 [[Header：[Session]]] 对象。

以下是我的示例代码：

enum TimelineItemType: String {  
  case Header = "header"   
  case Session = "session"
}

protocol TimelineItem {
  var id: Int { get }
  var type: TimelineItemType { get }
  var startDate: NSDate { get }
}

Header 结构体

struct Header: TimelineItem, Decodable, Hashable, Equatable {
  let id: Int
  let type: TimelineItemType = .Header
  let startDate: NSDate
  let text: String

  init?(json: JSON) {
    guard let id: Int = "id" <~~ json,
      let type: TimelineItemType = "type" <~~ json,
      let startDate: NSDate = "startDate" <~~ json,
      let text: String = "text" <~~ json where type == .Header else {
        return nil
    }
    self.id = id
    self.startDate = startDate
    self.text = text
  }

  var hashValue: Int {
    // As id is unique, we can use it for hash purpose
    return id
  }
}

Session 结构体

struct Session: TimelineItem, Decodable, Equatable {
  let id: Int
  let type: TimelineItemType = .Session
  let startDate: NSDate
  let name: String
  let syllabus: String
  let speaker: Speaker
  let language: String
  let room: String
  let duration: Int

  init?(json: JSON) {
    guard let id: Int = "id" <~~ json,
      let type: TimelineItemType = "type" <~~ json,
      let startDate: NSDate = Decoder.decodeDateISO8601("startDate")(json),
      let name: String = "name" <~~ json,
      let speaker: Speaker = "speaker" <~~ json,
      let syllabus: String = "syllabus" <~~ json,
      let language: String = "language" <~~ json,
      let room: String = "room" <~~ json,
      let duration: Int = "duration" <~~ json where type == .Session else {
        return nil
    }
    self.id = id
    self.startDate = startDate
    self.name = name
    self.speaker = speaker
    self.syllabus = syllabus
    self.language = language
    self.room = room
    self.duration = duration
  }
}

最后是我尝试分割数组的代码：

func timelineFromItems(timelineItems: [TimelineItem]) -> [[Header: [Session]]]? {

    let slicedSessions = timelineItems.split { $0 is Header }
    let sessions = Array(slicedSessions)
    let headers = timelineItems.filter { $0.type == .Header }
    var timeline = [[Header: [Session]]]()
    // HOW TO FILL THE TIMELINE ??
}

HOW TO FILL THE TIMELINE ?

- GUL

Answer 1

我通过删除所有不必要的信息来简化此示例。

protocol P {}
struct A: P, Hashable {
    var i:Int
    var hashValue: Int { return i }
}
func ==(lhs: A, rhs: A)->Bool {
    return lhs.i == rhs.i
}
struct B: P {
    var i:Int
}

// your current data
let arr:[P] = [A(i: 1),B(i: 1),B(i: 2), A(i: 2), B(i: 3), B(i: 4), B(i: 5)]

将您的数据转换为所需格式的函数

func foo(arr: [P])->[[A:[B]]]? {

    var dict:[A:[B]] = [:]
    var arrb:[B] = []
    let arrk:[A] = arr.filter { $0 is A }.map { $0 as! A }
    guard var key = arr[0] as? A else { return nil }

    arr.forEach { (p) in

        if let a = p as? A {
            dict[key] = arrb
            arrb = []
            key = a
        }
        if let b = p as? B {
            arrb.append(b)
        }
    }
    dict[key] = arrb
    var arrr:[[A:[B]]] = []
    arrk.forEach { (a) in
        if let arrb = dict[a] {
            arrr.append([a:arrb])
        }
    }
    return arrr
}

现在生成的数组符合您的要求（希望如此 :-)）。

if let result = foo(arr) {
    print(result) // [[A(i: 1): [B(i: 1), B(i: 2)]], [A(i: 2): [B(i: 3), B(i: 4), B(i: 5)]]]
}

另一个测试数据

let arr:[P] = [A(i: 1),B(i: 1),B(i: 2), A(i: 2), A(i: 3), B(i: 3)]

给你

[[A(i: 1): [B(i: 1), B(i: 2)]], [A(i: 2): []], [A(i: 3): [B(i: 3)]]]

所以，即使没有B跟随A，它也能正常工作。

- user3441734

Answer 2

我认为填充时间轴没有任何障碍，以下是我会做的方式：

func timelineFromItems(timelineItems: [TimelineItem]) -> [[Header: [Session]]]? {

    let slicedSessions = timelineItems.split { $0 is Header }
    let sessions = Array(slicedSessions)
    let headers = timelineItems.filter { $0.type == .Header }
    var timeline = [[Header: [Session]]]()

    if sessions.count == headers.count { // Check to be sure you have as much header as sessions
        for (index, value) in headers.enumerate() {
            let dictionary = [value: sessions.getElement(index)]
            timeline.append(dictionary)
        }
    }

    return timeline
}

- The Tom

---

for语句中缺少开头的{。 - GUL

我刚试了一下你的解决方案，它似乎非常接近我想要的。但是我遇到了一个编译错误：sessions[index]中出现了Ambiguous reference to subscript。 - GUL

你需要让 sessions = Array(slicedSessions) 吗？如果你将其删除并用 slicedSessions[index] 替换 sessions[index]，它还能正常工作吗？ - The Tom

“slicedSessions”也出现了同样的问题。 - GUL

或者可以尝试使用sessions.objectAtIndex(index)，不过我不确定在Swift中是否可行。 - The Tom

Answer 3

您可以使用以下简单的代码：

您实际上可以选择像这样的东西：

func deserialize(input: [TimelineItem]) -> [Header: [Session]] {
    var result = [Header: [Session]]()

    var latestHeader: Header! = nil

    input.forEach() {
        if let header = $0 as? Header {
            latestHeader = header
            result[header] = []
        } else if let session = $0 as? Session {
            result[latestHeader]!.append(session)
        }
    }

    return result
}

更新

虽然不太简单，但仍然简洁（上面的代码生成一个包含[Session]元素的字典，但是正如@user3441734指出的那样，你需要一个由单个元素字典组成的数组）:

func deserialize(input: [TimelineItem]) -> [[Header: [Session]]] {
    var result = [[Header: [Session]]]()

    var latestHeader: Header? = nil

    input.forEach() {
        if let header = $0 as? Header {
            latestHeader = header
            result.append([header: []])
        } else if let session = $0 as? Session, let header = latestHeader {
            var headerDict = result.popLast()!
            headerDict[header]!.append(session)
            result.append(headerDict)
        }
    }

    return result
}

- 0x416e746f6e

---

你的结果是字典，所以是无序集合。我的第一次尝试也是这样，直到我意识到他需要有序的集合/字典数组，其中字典只有一个键和一个关联值（数组）/... Array <[Header：[Session]]>。 - user3441734

哦，是的。我的错。所以他需要一个由单元素字典组成的数组。那我稍微调整一下代码。 - 0x416e746f6e

我做到了，看看我的答案。尽管代码“不是最好的”，但你可以看到它按预期工作 :-) - user3441734

现在它可以工作了！我的方法要干净得多！您能否从有效性的角度与我的两阶段方法进行比较？顺便说一句，如果由于某种原因TimeLineItem [0]不是Header，则结果会忽略它以及所有接下来的Session（直到下一个Header为止）...这可能没问题，我只是为了澄清而提出。 - user3441734

好的，是的。这段代码只是为了演示这个想法而草拟的。OP应该根据具体需求进行调整。 - 0x416e746f6e

Answer 4

感谢大家的帮助，我通过以下代码成功满足了我的需求：

```

typealias Section = [header: Header, sessions: [TimelineItem]]

private static func groupBySection(timelineItems: [TimelineItem]) -> [Section]? {
    let slicedSessions = timelineItems.split { $0.type == .Header }
    let headers = timelineItems.filter { $0.type == .Header }
    var timeline = [Section]()

    for (index, value) in headers.enumerate() {
      guard index < slicedSessions.count else {
        break
      }

      let sessionIndex = slicedSessions.startIndex.advancedBy(index)
      let arraySessions = Array(slicedSessions[sessionIndex])
      let header = value as! Header
      let section = (header: header, sessions: arraySessions)
      timeline.append(section)
    }
    return timeline.count > 0 ? timeline : nil
  }

```

- GUL