"""
Napiši funkcijo vsi_naslednji(vrata), ki prejme neko stanje vrat in vrne množico vseh možnih naslednjih stanj - torej vseh nizov, ki se od podanega razlikujejo na točno enem mestu. Klic vsi_naslednji('OO.O.') vrne množico {'OO.OO', 'OO...', 'OOOO.', 'O..O.', '.O.O.'}
"""
def ena_sprememba(vrata1, vrata2):
    stevec = 0
    for i in range(len(vrata1)):
        if (vrata1[i] != vrata2[i]):
            stevec += 1
    if (stevec == 1):
        return True
    return False

def vsi_naslednji(vrata):
    ad = set()
    a = set(itertools.permutations(vrata))
    for i in a:
        if ena_sprememba(i, vrata) is True:
            ad.add(i)
    return {i}