Nave - nivel

import copy

class Nod:
    def __init__(self, nrStare, stare):
        self.nrStare=nrStare
        self.stare=stare
        self.valoare=self.calculeazaValoare()

    def calculeazaValoare(self):
        valoare=50

        if len(self.stare)>4:
            if self.stare[4]==2:
                valoare+=4
        if len(self.stare)>2:
            if self.stare[2]==1:
                valoare+=5
        for i in range(0,len(self.stare)-1):
            if self.stare[i]%2-(self.stare[i+1]%2)==0:
                valoare-=8
        return valoare

    def navaDejaRepartizata(self, nava):
        if nava in self.stare:
            return True
        return False

def afisareSolutie(nod):
    print ("Repartizarea cea mai buna este (", end="")
    for echipaj in nod.stare:
        print ("{} ".format(echipaj), end="")
    print (") cu valoarea {}.".format(nod.valoare))

def Cautare():

    frontiera=[]
    teritoriu=[]
    optim=Nod(1,[])
    frontiera.append(optim)

    while len(frontiera)!=0:
        nodCurent=frontiera.pop(0)
        teritoriu.append(nodCurent)
        for i in range(1,6):
            if nodCurent.navaDejaRepartizata(i)==False:
                stare=copy.deepcopy(nodCurent.stare)
                stare.append(i)
                nod=Nod(len(frontiera)+len(teritoriu)+1,stare)
                if len(stare)==5 and nod.valoare>optim.valoare:
                    optim=nod
                frontiera.append(nod)

    afisareSolutie(optim)

    print ("Au fost expandate {} noduri.".format(len(teritoriu)))

Cautare()