Сортировка массива строк с несколькими условиями - PullRequest
Новая
       18

Сортировка массива строк с несколькими условиями

0 голосов
/ 03 мая 2019

У меня есть массив, SetDataList , который заполняется информацией ниже: 

["2019-04-30   Set 1:   8 Reps   55kg", "2019-05-02   Set 1:   5 Reps   70kg", "2019-05-02   Set 2:   4 Reps   70kg", "2019-05-02   Set 3:   2 Reps   75kg"]

В настоящее время я использую приведенный ниже код, чтобы упорядочить массив таким образом, чтобы множества, 1, 2 и 3 были сгруппированы и отображались как есть в приведенном выше массиве.

SetDataList.sort(by: {$0 < $1})

Однако я также хотел бы упорядочить массив в порядке убывания даты. Ранее я выполнял это с помощью следующего кода, хотя этот подход шифрует установленный порядок. 

let dateFormatter = DateFormatter()
dateFormatter.dateFormat = "yyyy-MM-dd"

let ordered = SetDataList.sorted { string1, string2 in
    guard let date1 = dateFormatter.date(from: string1), let date2 = dateFormatter.date(from: string2) else { return false }
    return date1 < date2
}

Как я могу реализовать несколько сортов? Спасибо

1 Ответ

0 голосов
/ 03 мая 2019

Я бы рекомендовал использовать структуры вместо строк, например: 

struct SetData {
    let setNumber: Int
    let date: Date
    let reps: Int
    let weight: Int
}

Давайте начнем с этого массива: 

let setDataList = ["2019-04-30   Set 1:   8 Reps   55kg",
                   "2019-05-02   Set 10:   20 Reps   70kg",
                   "2019-05-02   Set 1:   5 Reps   70kg",
                   "2019-05-02   Set 2:   4 Reps   70kg",
                   "2019-05-02   Set 3:   2 Reps   75kg"].shuffled()

Предположим, строки в массиве имеют одинаковый формат, вы можете отсортировать setDataList следующим образом: 

let sorted = setDataList.sorted { string1, string2 in
    let dateDelimiter = string1.index(string1.startIndex, offsetBy: 9)
    let earlierDate =  string1[...dateDelimiter] < string2[...dateDelimiter]

    if earlierDate { return true }

    let firstSetStart = string1.index(string1.startIndex, offsetBy: 17)
    var firstSetEnd = firstSetStart

    while firstSetEnd < string1.endIndex,
        string1[firstSetEnd] != ":" {
        firstSetEnd = string1.index(after: firstSetEnd)
    }

    let secondSetStart = string2.index(string2.startIndex, offsetBy: 17)
    var secondSetEnd = secondSetStart

    while secondSetEnd < string2.endIndex,
        string2[secondSetEnd] != ":" {
            secondSetEnd = string2.index(after: secondSetEnd)
    }

    let sub1 = string1[firstSetStart..<firstSetEnd]
    let sub2 = string2[secondSetStart..<secondSetEnd]

    guard let setNbr1 = Int(sub1), let setNbr2 = Int(sub2) else {
        fatalError("Wrong string structure")
    }

    return setNbr1 < setNbr2
}

И вы можете проверить результат следующим образом: 

for element in sorted {
    print(element)
}

Какие отпечатки: 

2019-04-30   Set 1:   8 Reps   55kg
2019-05-02   Set 1:   5 Reps   70kg
2019-05-02   Set 2:   4 Reps   70kg
2019-05-02   Set 3:   2 Reps   75kg
2019-05-02   Set 10:   20 Reps   70kg
...