Итерация по Vec с индексом в Rust: 4 способа сравнить

Итерация по Vec с индексом в Rust: от новичка до профи

Представьте: вы пишете на Rust и вам нужно пройтись по элементам вектора, но при этом знать не только значение, но и его позицию. Казалось бы, простая задача, но у новичков она часто вызывает ступор. Где тут for i in range(len(vec)), как в Python? А если я хочу менять элементы? А что быстрее?

В этой статье мы разберём все популярные способы итерации по Vec с индексом в Rust, сравним их производительность, читаемость и безопасность. Вы узнаете, какой метод выбрать в 2024 году и почему enumerate — это не всегда серебряная пуля.

Почему это важно?

В Rust нет классического for i in 0..vec.len() как единственно верного пути. Язык предлагает несколько парадигм: от функциональных итераторов до низкоуровневых циклов. Понимание разницы между ними — ключ к написанию эффективного и идиоматичного кода. Особенно это критично при работе с векторами, где индекс может понадобиться для модификации соседних элементов или отслеживания позиции в алгоритме.

Метод 1: Классика — enumerate

Самый распространённый и рекомендуемый способ. Метод enumerate оборачивает итератор, добавляя к каждому элементу его индекс:

let vec = vec!["a", "b", "c", "d"];
for (index, value) in vec.iter().enumerate() {
    println!("Позиция {}: {}", index, value);
}

Плюсы:

• Идиоматично и читаемо
• Безопасно — нет риска выхода за границы
• Работает с любым итератором, не только с Vec

Минусы: Нельзя напрямую изменять элементы вектора (нужен iter_mut()).

Метод 2: По индексам — 0..vec.len()

Подход, знакомый всем из C++ или Python. Используем диапазон индексов:

let mut vec = vec![10, 20, 30, 40];
for i in 0..vec.len() {
    vec[i] += i; // изменяем элемент по индексу
    println!("vec[{}] = {}", i, vec[i]);
}

Плюсы:

• Полный контроль над индексом и элементом
• Можно менять элементы через vec[i]
• Интуитивно понятен новичкам

Минусы: Медленнее, чем итераторы (дополнительная проверка границ). Rust-сообщество считает его "не идиоматичным".

Метод 3: while с счётчиком

Самый "ручной" способ, который вы встретите в старом коде или при оптимизации:

let vec = vec![1.0, 2.0, 3.0];
let mut i = 0;
while i < vec.len() {
    println!("Элемент {}: {}", i, vec[i]);
    i += 1;
}

Плюсы: Максимальная гибкость (можно прыгать по индексам, менять шаг).

Минусы: Легко ошибиться с условием. Не рекомендуется без веской причины.

Метод 4: zip с диапазоном

Функциональный подход, когда нужно итерировать два вектора параллельно с индексами:

let names = vec!["Alice", "Bob", "Charlie"];
let scores = vec![95, 87, 92];

for (i, (name, score)) in (0..).zip(names.iter()).zip(scores.iter()) {
    println!("{}: {} набрал {} очков", i, name, score);
}

Плюсы: Мощно для параллельной итерации. Бесконечный диапазон 0.. автоматически обрезается.

Минусы: Может быть сложен для чтения. Избыточен для простых задач.

Сравнение производительности

Мы прогнали бенчмарки на векторе из 100 000 элементов (оптимизация --release):

Вывод: Разница не критична для 99% приложений. Итераторы (enumerate) не только читаемее, но и чуть быстрее благодаря оптимизациям компилятора (LLVM).

Когда какой метод выбрать?

• Нужен только индекс и значение? — enumerate. Всегда.
• Надо изменять элементы по индексу? — iter_mut().enumerate() или 0..vec.len(). Первый вариант безопаснее.
• Пишем высокопроизводительный код? — Используйте enumerate, но проверьте бенчмарки под вашу нагрузку.
• Итерируем несколько векторов? — zip с диапазоном или enumerate на одном из них.

Продвинутый совет: изменяем элементы с enumerate

Многие новички думают, что enumerate не позволяет изменять вектор. Это не так — просто используйте iter_mut():

let mut vec = vec![1, 2, 3, 4, 5];
for (i, val) in vec.iter_mut().enumerate() {
    *val *= i + 1; // умножаем каждый элемент на его позицию+1
}
println!("{:?}", vec); // [1, 4, 9, 16, 25]

Типичные ошибки

1. Забыть .iter(): for (i, v) in vec.enumerate() не скомпилируется — Vec сам по себе не является итератором.
2. Использовать for i in 0..vec.len() при неизменяемом доступе: Работает, но менее идиоматично.
3. Пытаться получить индекс в for v in &vec: Индекс не доступен напрямую.

Заключение: ваш идеальный выбор