Как найти индекс элемента в списке (массиве) в Swift?

Использование метода firstIndex(of:)

В Swift для поиска индекса элемента в массиве существует встроенный метод firstIndex(of:). Он возвращает опциональное значение (Int?): если элемент найден — вы получите его индекс, если нет — nil. Это самый простой и правильный способ.

Вот как это работает с вашим примером:

let users = ["Анна", "Борис", "Виктор", "Галина", "Дмитрий"]

if let index = users.firstIndex(of: "Виктор") {
    print("Элемент найден на индексе: \(index)")
} else {
    print("Элемент не найден")
}
// Вывод: "Элемент найден на индексе: 2"

Обратите внимание на конструкцию if let — это безопасное извлечение опционала. Она гарантирует, что код внутри блока if выполнится только если индекс действительно существует.

Что делать, если элемент повторяется?

Метод firstIndex(of:) возвращает только первое вхождение. Если вам нужны все индексы, используйте enumerated() в комбинации с filter:

let numbers = [1, 2, 3, 2, 4, 2, 5]
let target = 2

let allIndices = numbers.enumerated()
    .filter { $0.element == target }
    .map { $0.offset }

print(allIndices) // [1, 3, 5]

Поиск по условию (для сложных типов)

Если у вас массив не простых значений (например, структуры или классы), используйте firstIndex(where:). Он принимает замыкание с условием:

struct User {
    let name: String
    let age: Int
}

let people = [
    User(name: "Анна", age: 25),
    User(name: "Борис", age: 30),
    User(name: "Виктор", age: 35)
]

if let index = people.firstIndex(where: { $0.name == "Виктор" }) {
    print("Пользователь Виктор на индексе \(index)")
}
// Вывод: "Пользователь Виктор на индексе 2"

Производительность и best practices

• Сложность: firstIndex(of:) работает за O(n) — в худшем случае метод просматривает весь массив. Для большинства задач это приемлемо.
• Проверка существования: Если вам нужно только узнать, есть ли элемент в массиве (без индекса), используйте contains(_:) — он читается понятнее.
• Для уникальных элементов: Если вы часто ищете индексы, рассмотрите использование словаря (Dictionary) для быстрого доступа по ключу.
• Безопасность: Всегда обрабатывайте опциональный результат. Прямое принудительное извлечение (users.firstIndex(of: "Кто-то")!) приведёт к крашу, если элемента нет.

Ваш ручной цикл — это абсолютно рабочий подход, но встроенные методы делают код короче, понятнее и менее подверженным ошибкам. Используйте firstIndex(of:) для простых значений и firstIndex(where:) для сложных условий.

extension Array where Element: Equatable {
    func allIndices(of element: Element) -> [Int] {
        return self.enumerated()
            .compactMap { $0.element == element ? $0.offset : nil }
    }
}

let items = ["a", "b", "a", "c", "a"]
let indices = items.allIndices(of: "a")
print(indices) // [0, 2, 4]

let optionalNames: [String?] = ["Анна", nil, "Виктор", nil]
if let index = optionalNames.firstIndex(where: { $0 == "Виктор" }) {
    print("Найдено на индексе \(index)")
}