Implementación de BFS en Árboles usando Python

Clase 21 de 52Curso de Algoritmos Avanzados: Grafos y Árboles

Resumen

¿Cómo implementar BFS para recorrer un árbol?

El algoritmo de Búsqueda en Anchura (BFS, por sus siglas en inglés) es una técnica popular para explorar o recorrer estructuras de datos, como árboles o grafos. En este caso, aprenderemos cómo implementar BFS en un árbol y cómo imprimir todos sus valores de manera eficiente utilizando colas. Este enfoque es esencial para cualquier programador y puede aplicarse en diversos lenguajes de programación.

¿Cómo se configura la función BFS?

Primero, definimos una función que llamaremos BFS, la cual recibirá como entrada el nodo raíz del árbol. El objetivo es retornar una lista con los nodos organizados en el orden que BFS precisa. Aquí está cómo se estructura:

def BFS(raiz):
    if not raiz:
        return []  # Si no hay raíz, simplemente retornamos una lista vacía
    
    respuesta = []  # Lista donde guardaremos el orden de los nodos
    cola = [raiz]  # Iniciamos la cola con el nodo raíz
    
    nivel = 0  # Contador de niveles del árbol
    while cola:  # Mientras haya elementos en la cola
        nivel += 1
        for _ in range(len(cola)):
            nodo_actual = cola.pop(0)  # Sacamos el primer elemento de la cola
            # Aquí se procesa el nodo_actual si es necesario
            respuesta.append(nodo_actual)  # Añadimos el nodo actual a la respuesta
            # Añadimos los hijos del nodo actual a la cola
            for hija in nodo_actual.hijas:
                cola.append(hija)

    print("La cantidad de niveles de este árbol es", nivel)
    return respuesta

¿Cómo se trata el recorrido del árbol por niveles?

Durante la ejecución del algoritmo, se maneja un contador de nivel que incrementa cada vez que completamos un nivel del árbol. Esto es parte fundamental para entender la profundidad del árbol recorrido por BFS.

  1. Inicialización de la cola: Se comienza la cola con el nodo raíz.
  2. Iteración por nivel: Mientras existan nodos en la cola, se incrementa el contador de nivel y se procesa cada nodo.
  3. Procesamiento de nodos: Se extrae cada nodo de la cola, se procesa (si es necesario) y se agregan sus hijos a la cola, para así considerar el siguiente nivel.

¿Cómo se manejan los hijos de un nodo?

En este algoritmo, se supone inicialmente que los nodos podrías tener múltiples hijos almacenados en una lista llamada hijas. Este modelo se adapta a árboles n-arios, sin embargo, para árboles binarios, podrías manejarlo directamente mediante propiedades izquierda y derecha.

Ejemplo de manejo de hijos en un árbol binario:

if nodo_actual.izquierda:
    cola.append(nodo_actual.izquierda)
if nodo_actual.derecha:
    cola.append(nodo_actual.derecha)

Esta modificación permite adaptar BFS a cualquier tipo de árbol, siendo flexible y eficiente.

¿Qué hacemos al finalizar la iteración del árbol?

Una vez que el algoritmo ha recorrido todos los niveles y procesado cada nodo, puedes realizar diversas tareas como imprimir el resultado. En este caso, el BFS completo resultará en una lista respuesta que contiene los nodos en el orden específico de búsqueda en anchura.

El método BFS es un paradigma fundamental en informática para el manejo y procesamiento de estructuras de datos complejas. ¡Esperamos que estos conocimientos fortalezcan tus habilidades y te motiven a seguir explorando otros algoritmos esenciales!