swift - 在名称数组中添加像字母一样的地址簿

Question

我正在尝试选取一组人(目前仅用名字来简化)，不仅对其进行排序，而且还在以该字母开头的所有人员前面添加第一个字母。

基本上是这样的:[“Daniel”，“Michael”，“Lisa”，“Lena”]应该变成这样:[“D”，“Daniel”，“L”，“Lena”，“Lisa”，“M ”，“迈克尔”]

我在 Playground 上有一些东西正在运行，但对我来说，这看起来很丑陋，并且不知道如果人们达到数千或更多，性能会如何。

//: Playground - noun: a place where people can play

import Cocoa

var previousName: String?

let people = ["Martin", "Michael", "Lisa", "Lena", "Dirk"]
var peopleWithLetters: [String] = []

people.sorted { $0.localizedCaseInsensitiveCompare($1) == ComparisonResult.orderedAscending }.forEach {
    if $0.uppercased().characters.first! != previousName?.uppercased().characters.first! {
        peopleWithLetters.append("\($0.uppercased().characters.first!)")
    }
    previousName = $0
    peopleWithLetters.append($0)
}

任何关于如何简化或遵循最佳实践的想法都将受到高度赞赏。

Answer 1

这是函数式编程大放异彩的完美用例。

• 将名称数组映射到首字母数组
• 使用集合删除重复项
• 将字母与名称组合
• 对数组进行排序

这是代码:

//: Playground - noun: a place where people can play

func addSections(in words: [String]) -> [String] {
    let sections = Set(words.map { String($0[$0.startIndex]) })
    let wordsWithSections = words + sections
    return wordsWithSections.sorted { $0 < $1 }
}

let names = ["Daniel", "Michael", "Lisa", "Lena"]
addSections(in: names)

尽管如果您计划使用此数组来提供 tableView(部分/行)，那么最好使用您在评论中提到的字典。