Перебор с itertools

Модуль itertools генерирует сочетания и упорядочивания за вас, поэтому вам нужно писать меньше циклов и можно допустить меньше ошибок. 🙂

Всё находится в одном модуле, поэтому начните с импорта itertools. Затем можно напрямую вызывать product, permutations и combinations.

Используйте product, чтобы заменить глубоко вложенные циклы по одним и тем же вариантам. Он выдаёт каждый кортеж выбранных значений из заданных диапазонов.

Когда порядок важен, permutations перечисляет все расположения элементов. Это идеально подходит для полного перебора небольших задач в стиле задачи коммивояжёра.

Когда порядок не важен, combinations выбирает каждую группу размера r. Это отлично подходит для выбора подмножества элементов для проверки.

Ключевое различие — порядок: permutations считает (1,2) и (2,1) разными вариантами, а combinations считает их одним и тем же выбором.

Количество перестановок растёт как n!, то есть очень быстро увеличивается. Даже при n, равном 11, оно уже превышает сорок миллионов вариантов, поэтому значение n должно быть небольшим.

Эти функции являются генераторами: они выдают по одному элементу за раз, не создавая огромный список, поэтому при переборе расход памяти остаётся небольшим.

Сначала сгенерируйте кандидатов, а затем оставьте только те, которые проходят вашу проверку. Этот шаблон превращает сложный поиск в короткий и понятный цикл.

Нужно выбирать элементы, которые могут повторяться? Используйте сочетания с повторениями, чтобы один и тот же элемент мог входить в группу несколько раз.

Задайте два вопроса: важен ли порядок и могут ли элементы повторяться? Ответы сразу укажут на правильную функцию itertools.

Вам нужно перечислить все способы выбрать 3 элемента из 6, при этом порядок не важен.

Используйте product, permutations и combinations для аккуратного перечисления вариантов. Выбирайте инструмент, выяснив, важен ли порядок и повторяются ли элементы, и следите за ростом количества вариантов. 🚀