Ir al contenido
wannacode

Una tabla hash como contador

En resumen: Un mapa usado como contador (dict o collections.Counter en Python) cuenta cuántas veces aparece cada elemento en una sola pasada: O(n) en lugar de O(n²) al buscar en una lista.

Problema

Los productos más vendidos del día

El panel de la tienda muestra los tres productos más vendidos del día. El registro es una lista de pedidos, uno por línea. El código mantiene dos listas, los nombres y los contadores, y para cada pedido busca el producto desde el principio de los nombres:

log = log_of(1000)
names = []
counts = []
comparisons = 0
for item in log:
i = 0
while i < len(names):
comparisons += 1
if names[i] == item:
break
i += 1
if i == len(names):
names.append(item)
counts.append(1)
else:
counts[i] += 1
print(f"pedidos: {len(log)}")
print(f"productos: {len(names)}")
print(f"comparaciones: {comparisons}")
Salida
pedidos: 1000
productos: 251
comparaciones: 125255

Mil pedidos y 251 productos distintos: 125 255 comparaciones. El catálogo crece con la tienda, y con él la lista que recorre cada pedido. En un día grande con 100 000 pedidos el panel tarda minutos en cargar.

De dónde viene el registro

El registro sigue una fórmula, para que los programas en los cuatro lenguajes y la comparación del navegador de más abajo construyan exactamente el mismo. El catálogo tiene n / 4 + 1 productos:

def log_of(n):
catalog = n // 4 + 1
return [f"SKU-{(i * 7919) % catalog}" for i in range(n)]

La idea

El mapa cuenta

En la lección «dict y set» contaste pedidos con un mapa. Aquí se convierte en una técnica: una tabla hash como contador (un mapa de frecuencias). La clave es lo que cuentas, el valor es cuántas veces lo has visto.

  • Una pasada por los datos. Para cada elemento: encontrar su contador (o 0) y sumar 1.
  • La búsqueda se hace en una tabla hash: O(1) en promedio, sean cuales sean los productos. En total O(n) en lugar de O(n · m).
  • Después respondes las preguntas con los contadores, no con los datos: cuál es el más frecuente, cuál aparece una vez, si dos cosas tienen los mismos contadores.

La memoria es O(m): un par por cada producto distinto.

Míralo

El contador se llena

from collections import defaultdict
log = [
"ratón", "audífonos", "ratón", "monitor",
"teclado", "ratón", "audífonos", "monitor",
"ratón", "audífonos",
]
counts = defaultdict(int)
for item in log:
counts[item] += 1
top = sorted(counts.items(),
key=lambda e: e[1], reverse=True)
for item, n in top[:3]:
print(item, n)
Salida
ratón 4
audífonos 3
monitor 2

counts = defaultdict(int)
for item in log:
counts[item] += 1
top = sorted(counts.items(),
key=lambda e: e[1], reverse=True)
for item, n in top[:3]:
print(item, n)
Paso a paso, en palabras
  1. Un mapa vacío: producto → cuántas veces se vendió. El registro tiene 10 pedidos.
  2. Pedido 1: ratón. Todavía no está en el mapa, así que recibe un contador de 1.
  3. Pedido 2: audífonos. Todavía no está en el mapa, así que recibe un contador de 1.
  4. Pedido 3: ratón, 1 → 2. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  5. Pedido 4: monitor. Todavía no está en el mapa, así que recibe un contador de 1.
  6. Pedido 5: teclado. Todavía no está en el mapa, así que recibe un contador de 1.
  7. Pedido 6: ratón, 2 → 3. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  8. Pedido 7: audífonos, 1 → 2. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  9. Pedido 8: monitor, 1 → 2. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  10. Pedido 9: ratón, 3 → 4. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  11. Pedido 10: audífonos, 2 → 3. Una búsqueda en la tabla hash, sin recorrer los demás productos.
  12. 10 pedidos, 10 actualizaciones del contador. Se ordenan solo los 4 productos distintos por contador, no los pedidos.
  13. Los tres primeros: ratón 4, audífonos 3, monitor 2.

defaultdict(int) crea un contador en 0 la primera vez que se lee una clave nueva, así que counts[item] += 1 también funciona para un producto nuevo. Los tres primeros salen de ordenar los pares por contador: se ordenan solo los productos distintos, no los pedidos.

Pruébalo

Completa

La línea que cuenta:

counts = defaultdict()

Míralo

Las herramientas de tu lenguaje

Python tiene un contador ya hecho: collections.Counter. Toma la lista y cuenta por sí solo, da 0 para una clave que falta, y most_common(k) devuelve los k primeros por contador. Los contadores iguales se quedan en el orden de la primera aparición:

from collections import Counter
log = [
"ratón", "audífonos", "ratón", "monitor",
"teclado", "ratón", "audífonos", "monitor",
"ratón", "audífonos",
]
counts = Counter(log)
print(counts["ratón"], counts["tableta"])
print(counts.most_common(3))
Salida
4 0
[('ratón', 4), ('audífonos', 3), ('monitor', 2)]

defaultdict(int) es la herramienta más general: el mismo contador, pero el valor también puede ser una suma o una lista.

Míralo

Mídelo: una lista de pares frente a un mapa

Los mismos mil pedidos con el mapa:

log = log_of(1000)
counts = defaultdict(int)
updates = 0
for item in log:
counts[item] += 1
updates += 1
print(f"pedidos: {len(log)}")
print(f"productos: {len(counts)}")
print(f"actualizaciones del contador: {updates}")
Salida
pedidos: 1000
productos: 251
actualizaciones del contador: 1000

1000 actualizaciones en lugar de 125 255 comparaciones. Ejecuta las dos en el navegador. n es el número de pedidos, y el catálogo crece con él:

  • Una lista de pares de producto y cantidad O(n²)
  • Un mapa como contador O(n)
Algoritmon = 10n = 1000n = 100.000
Una lista de pares de producto y cantidad16125.255≈ 1.252.550.000
Un mapa como contador101000100.000
El código se ejecuta en tu navegador, en JavaScript. La cantidad de operaciones es la misma en todos los lenguajes, porque el algoritmo es el mismo. El tiempo depende de la máquina, así que solo sirve para comparar. «≈» es una estimación a partir de la curva: un n tan grande no lo ejecutamos, para que la página no se congele.
counts = defaultdict(int)
for item in log:
counts[item] += 1
top = sorted(counts.items(),
key=lambda e: e[1], reverse=True)
for item, n in top[:3]:
print(item, n)
Más precisamente, para curiosos

Los k primeros sin ordenarlo todo. Ordenar m productos distintos es O(m log m). Si hay millones de productos y necesitas los 10 primeros, mantienes un heap con como máximo k pares: cada par entra, y cuando hay k + 1 sale el más pequeño. Eso es O(m log k).

Una clave con varias partes. Cuentas ventas por ciudad y producto: la clave es el par. En Python es una tupla (city, item): es inmutable y tiene un hash.

La idea

El carrito y el kit

El kit «oficina en casa» es un monitor, un teclado, un ratón y dos cables. El almacén comprueba que el carrito que se va a enviar es exactamente el kit: nada de más, nada de menos. El orden no importa, los contadores sí.

Un contador: +1 por cada producto del kit, −1 por cada producto del carrito. El carrito coincide si todos los contadores terminan en 0.

def matches(kit, basket):
return Counter(kit) == Counter(basket)
Un Counter se compara con == por sus contadores, así que toda la comprobación es una línea.

La misma comprobación encuentra anagramas: «listen» y «silent» tienen las mismas letras con los mismos contadores. El contador es sobre letras en lugar de productos.

Pruébalo

Predice la comprobación

from collections import Counter
kit = ["monitor", "teclado", "ratón", "cable", "cable"]
basket1 = ["cable", "ratón", "monitor", "cable", "teclado"]
basket2 = ["monitor", "teclado", "ratón", "ratón", "cable"]
print(f"carrito 1: {'kit completo' if matches(kit, basket1) else 'no coincide'}")
print(f"carrito 2: {'kit completo' if matches(kit, basket2) else 'no coincide'}")
Respuesta

La idea

El primer cliente con un solo pedido

Marketing envía un vale de «vuelve» a los clientes que pidieron una sola vez. Quiere al primero en el tiempo. El contador da los contadores, pero no el orden:

def first_single(customers):
counts = Counter(customers)
for c in customers:
if counts[c] == 1:
return c
return None
from collections import Counter
customers = [
"C-17", "C-4", "C-17", "C-9",
"C-4", "C-21", "C-9", "C-30",
]
print(f"primer cliente con un solo pedido: {first_single(customers)}")
Salida
primer cliente con un solo pedido: C-21

Dos pasadas: la primera cuenta, la segunda recorre el registro, en el orden de los pedidos, y se detiene en el primero con contador 1. O(n), sin bucle anidado.

Más precisamente, para curiosos

¿Por qué la segunda pasada recorre el registro y no el mapa? Un dict guarda el orden de inserción (garantizado desde Python 3.7), así que aquí recorrer counts daría la misma respuesta. El registro sigue siendo más claro: el orden viene de él.

La idea

Dos productos por exactamente 50 €

El cliente tiene un vale de exactamente 50 € y quiere dos productos que lo agoten. Los precios del catálogo no están ordenados.

Recorres los precios y para cada uno preguntas: ¿he visto ya el precio que completa este hasta 50? El mapa guarda los precios vistos hasta ahora y sus posiciones:

def voucher_pair(prices, voucher):
seen = {}
for i, price in enumerate(prices):
j = seen.get(voucher - price)
if j is not None:
return j, i
seen[price] = i
return None
prices = [27, 14, 33, 25, 8, 36, 19]
a, b = voucher_pair(prices, 50)
print(f"productos {a} y {b}: {prices[a]} € + {prices[b]} €")
Salida
productos 1 y 5: 14 € + 36 €

Una pasada: O(n) de tiempo y O(n) de memoria. Con dos punteros funciona sin mapa, con memoria O(1), pero los precios tienen que estar ordenados. Si no lo están, ordenar cuesta O(n log n) y se pierde la posición de cada producto en el catálogo. Así que: datos ordenados, dos punteros; datos sin ordenar, un mapa.

Pruébalo

Encuentra el error

Alguien intercambió dos líneas. Ahora, para un vale de 50 €, el programa ofrece un producto dos veces:

prices = [27, 14, 33, 25, 8, 36, 19]
a, b = voucher_pair(prices, 50)
print(f"productos {a} y {b}: {prices[a]} € + {prices[b]} €")
Salida
productos 3 y 3: 25 € + 25 €

Elige la línea con el error.

Ejercicio

Quién gasta más

Resuelve la tarea en tu propio editor.

Ingresos por cliente. El registro tiene pares de cliente e importe: C-4 con 30 €, C-17 con 55 €, C-4 con 20 €, C-9 con 80 €, C-17 con 10 € y C-4 con 45 €. Suma los importes por cliente con un mapa (el valor es una suma, no un contador) e imprime los dos que más gastaron:

Salida
C-4: 95 €
C-9: 80 €

Más tareas

  • Anagramas. Escribe una comprobación de si dos palabras son anagramas («listen» y «silent») contando las letras con un mapa. ¿Cuántas operaciones necesita para palabras de longitud n?
  • La primera repetición. Encuentra el primer pedido cuyo cliente ya había pedido antes. ¿Necesitas un contador o basta un conjunto?
  • Periodos con contadores iguales. Con las sumas prefijas sabes contar periodos con una suma dada. Cuenta los periodos de días consecutivos en los que el número de pedidos devueltos es igual al de entregados.

Test

Ponte a prueba

1. El registro tiene 100 000 pedidos y 25 000 productos distintos. ¿Cuántas actualizaciones del contador hace el mapa?
2. ¿Cómo compruebas con un mapa que el carrito es exactamente el kit?
3. En voucherPair el precio entra en el mapa después de buscar su pareja. ¿Por qué no antes?
4. ¿Cuándo son mejores dos punteros para una pareja con una suma dada?
5. Necesitas los 10 primeros de 2 000 000 de productos distintos. ¿Qué es lo más barato?

Creado por Ivo Gergov como parte de la campaña de Evo Tech por una educación accesible en programación ·