Árbol¶
In [1]:
# FUNCIONES
def sa(a,b):
if len(a) > 1: a[0],a[1] = a[1],a[0]
return a,b
def sb(a,b):
if len(b) > 1: b[0],b[1] = b[1],b[0]
return a,b
def ss(a,b):
sa(a,b)
sb(a,b)
return a,b
def pa(a,b):
if len(b) > 0:
a.insert(0, b[0])
b.pop(0)
return a,b
def pb(a,b):
if len(a) > 0:
b.insert(0, a[0])
a.pop(0)
return a,b
def ra(a,b):
if len(a) > 1: a.append(a.pop(0))
return a,b
def rb(a,b):
if len(b) > 1: b.append(b.pop(0))
return a,b
def rr(a,b):
ra(a,b)
rb(a,b)
return a,b
def rra(a,b):
if len(a) > 1: a.insert(0, a.pop())
return a,b
def rrb(a,b):
if len(b) > 1: b.insert(0, b.pop())
return a,b
def rrr(a,b):
rra(a,b)
rrb(a,b)
return a,b
In [21]:
def separa_valor(v):
posicion = v.find('|')
a = v[:posicion]
b = v[posicion+1:]
a = a.replace(" ", "")
b = b.replace(" ", "")
a = [int(x) for x in a]
b = [int(y) for y in b]
return a, b
def agrega(k, instruccion, aCopy, bCopy):
pareja = ' '.join([str(x) for x in aCopy]) + "|" + ' '.join([str(x) for x in bCopy]) # '3 4 2 1|'
tupla = tuple([str(instruccion)])
if pareja not in d.values():
d[k+tupla] = pareja
def busca(a,b): # Estas son a y b: [4, 3, 2, 1] []
global d
while valor_ordenado not in d.values():
nivel = max([len(k) for k in d.keys()]) # el nivel del árbol es igual a la máxima longitud de las tuplas que van como clave en el diccionario. Inicialmente el nivel=0
d_nivel = {} # inicializamos un diccionario que copia d pero solo los items que sean del nivel máximo en ese momento
for k,v in d.items():
if len(k) == nivel:
d_nivel[k] = v
for k,v in d_nivel.items():
a, b = separa_valor(v)
for i in moves:
if (i == "sa"): aCopy, bCopy = sa(a[:], b[:])
elif (i == "sb"): aCopy, bCopy = sb(a[:], b[:])
elif (i == "ss"): aCopy, bCopy = ss(a[:], b[:])
elif (i == "pa"): aCopy, bCopy = pa(a[:], b[:])
elif (i == "pb"): aCopy, bCopy = pb(a[:], b[:])
elif (i == "ra"): aCopy, bCopy = ra(a[:], b[:])
elif (i == "rb"): aCopy, bCopy = rb(a[:], b[:])
elif (i == "rr"): aCopy, bCopy = rr(a[:], b[:])
elif (i == "rra"): aCopy, bCopy = rra(a[:], b[:])
elif (i == "rrb"): aCopy, bCopy = rrb(a[:], b[:])
elif (i == "rrr"): aCopy, bCopy = rrr(a[:], b[:])
agrega(k, i, aCopy, bCopy)
if __name__ == "__main__":
a_string = "6 5 4 3 2 1"
a = a_string.replace(" ", "") # '4321'
d = {(): a_string + "|"} # diccionario. Ejemplo: {():'4 3 2 1|', ('sa',):'3 4 2 1|', ('pb',):'3 2 1|4', ('ra',):'3 2 1 4|', ('rra',):'1 4 3 2|', ('pb', 'sa'):'2 3 1|4', ('pb', 'pb'):'2 1|3 4'}
moves = ["ra","rb","rra","rrb","pa","pb","sa","sb","ss","rr","rrr"]
a = [int(x) for x in a] # [4, 3, 2, 1]
b = []
valor_ordenado = ' '.join(map(str, sorted(a))) + '|' # "1 2 3 4|"
busca(a,b)
print(list(d.keys())[list(d.values()).index(valor_ordenado)]) # da la clave que corresponde al valor ordenado "1 2 3 4|"
#print(d)
('rra', 'rra', 'pb', 'pb', 'sa', 'ra', 'ra', 'ss', 'pa', 'pa')
In [3]:
!pip install openpyxl
Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/ Requirement already satisfied: openpyxl in /usr/local/lib/python3.8/dist-packages (3.0.10) Requirement already satisfied: et-xmlfile in /usr/local/lib/python3.8/dist-packages (from openpyxl) (1.1.0)
In [4]:
import openpyxl # importamos la librería openpyxl
wb = openpyxl.Workbook() # para crear por primera vez un libro
ws = wb.active # estas dos lineas crean el libro, aún sin nombre
wb.save('arbol.xlsx') # grabamos el fichero por primera vez
In [5]:
# Cuando ya sebemos que queremos trabajar con el archivo arbol.xlsx
wb = openpyxl.load_workbook('arbol.xlsx')
ws = wb["Sheet"] # accediendo a la hoja Sheet
fila = 11
for k,v in d.items():
ws.cell(row=fila, column=3).value = str(k)
ws.cell(row=fila, column=4).value = v
fila += 1
wb.save('arbol.xlsx')
wb.close()