Python: Деревья
2026-02-26 18:59 Diff
https://ru.hexlet.io/courses/python-trees/lessons/accumulator/theory_unit

В некоторых ситуациях во время обхода дерева нужна дополнительная информация, которая зависит от расположения узла. Ее невозможно получить из описания самого узла, так как узел ее не содержит. Эту информацию нужно собирать прямо во время обхода.

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

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

Как работать с аккумулятором

Возьмем для примера такую задачу — найдем все пустые директории в нашей файловой системе.

Сначала реализуем простую версию, затем усложним ее и внедрим аккумулятор. Посмотрим на пример файловой системы:

В этой структуре три пустых директории:

  • /logs
  • /etc/apache
  • /etc/consul/data

Код, решающий эту задачу, выглядит так:

Самое необычное в этой реализации — это функция flatten(). Далее обсудим, зачем она нужна.

Если оставить только map(), то результат может удивить:

Такое происходит из-за возврата списка на каждом уровне вложенности. На выходе получается список списков, напоминающий по структуре исходное файловое дерево. Чтобы этого не происходило, нужно выправлять код с помощью flatten().

Попробуем усложнить задачу. Найдем все пустые директории, но с максимальной глубиной поиска в два уровня. Например, директории /logs и /etc/apache подходят под это условие, а вот /etc/consul/data — нет.

Для начала нужно понять, откуда брать глубину. В деревьях глубина считается как количество ребер от корня до нужного узла. Визуально ее посчитать легко, а что насчет кода? Глубину конкретного узла можно представить как глубину предыдущего узла плюс единица.

Следующий шаг – добавить переменную, которая передается при каждом рекурсивном вызове (проваливающимся в директорию). В нашем случае эта переменная содержит текущую глубину. На каждом уровне внутри каждой директории к ней добавляется единица.

Такую переменную называют аккумулятором — она аккумулирует, то есть накапливает данные.

Единственная проблема заключается в том, что у исходной функции find_empty_dirs() ровно один параметр – это узел. С помощью этой функции невозможно передавать глубину всем вложенным директориям и файлам.

Поэтому придется ввести внутреннюю функцию, которая сможет пробрасывать аккумулятор дальше по дереву:

Можно пойти еще дальше и позволить указывать максимальную глубину снаружи:

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

Правильный способ сделать это – использовать в качестве значения по умолчанию бесконечность math.inf: