Editing 2588 - Iepuras1 (section)

= Exemplu1: =
<code>iepuras1.in</code>
 10 15
 7 1 4
 0 0 6 0 5 5 0 7 0 0 0 0 0 0 0
 0 0 0 0 5 5 0 0 4 0 0 0 0 0 3
 0 5 5 0 5 5 0 0 0 0 0 0 0 0 0 
 0 5 5 0 5 5 0 1 7 0 0 7 0 0 2
 0 0 0 0 5 5 0 0 0 0 0 0 0 0 0
 0 0 0 0 0 5 0 0 0 0 0 0 0 0 0
 0 3 0 0 0 5 6 0 0 5 5 5 0 0 0
 0 7 0 0 0 0 0 0 0 0 0 0 0 5 0
 0 4 0 0 0 7 0 0 3 5 0 0 0 5 0
 0 6 0 0 0 0 0 0 1 5 0 0 0 5 0
 0 
<code>iepuras1.out</code>
 FERICIT
 1 8
 5 37
<syntaxhighlight lang="python" line="1">
def simulare_gradina(m, n, gradina, poz_initiala, directie_initiala):
    # Direcțiile de mișcare: nord, vest, sud, est
    directii = {
        'nord': (-1, 0),
        'vest': (0, -1),
        'sud': (1, 0),
        'est': (0, 1)
    }
    
    # Mapare coduri săgeți la direcții
    sageata_la_directie = {
        1: 'nord',
        2: 'vest',
        3: 'sud',
        4: 'est'
    }
    
    # Poziția curentă și direcția curentă
    x, y = poz_initiala
    directie = directie_initiala
    
    # Contoare
    morcovi_ronțăiți = 0
    pasi = 0
    treceri_pe_langa_bombe = {}
    starea_finala = 'rătăcit'  # implicit
    
    # Funcție pentru verificarea granițelor
    def in_gradina(x, y):
        return 0 <= x < m and 0 <= y < n
    
    # Pozițiile inițiale ale zidurilor
    ziduri_initiale = set()
    for i in range(m):
        for j in range(n):
            if gradina[i][j] == 8:
                ziduri_initiale.add((i, j))
    
    # Pozițiile zidurilor alternative
    ziduri_alternative = set()
    
    # Inițial, zidurile sunt cele inițiale
    ziduri_curente = ziduri_initiale.copy()
    
    # Traseul iepurașului
    while True:
        if not in_gradina(x, y):
            starea_finala = 'rătăcit'
            break
        
        pasi += 1
        celula = gradina[x][y]
        
        if celula == 7:  # morcov
            morcovi_ronțăiți += 1
            gradina[x][y] = 0
        
        elif celula in sageata_la_directie:  # săgeată
            directie = sageata_la_directie[celula]
            gradina[x][y] = 0
        
        elif celula == 5:  # pom
            dx, dy = directii[directie]
            x -= dx
            y -= dy
            directie = {'nord': 'sud', 'sud': 'nord', 'vest': 'est', 'est': 'vest'}[directie]
        
        elif celula == 6:  # bombă
            starea_finala = 'îngeraș'
            break
        
        elif celula == 9:  # trapă
            ziduri_curente = ziduri_alternative if ziduri_curente == ziduri_initiale else ziduri_initiale
        
        # Verificăm dacă trece pe lângă o bombă
        vecini = [(x-1, y), (x+1, y), (x, y-1), (x, y+1)]
        for vx, vy in vecini:
            if in_gradina(vx, vy) and gradina[vx][vy] == 6:
                if (vx, vy) not in treceri_pe_langa_bombe:
                    treceri_pe_langa_bombe[(vx, vy)] = 0
                treceri_pe_langa_bombe[(vx, vy)] += 1
                if treceri_pe_langa_bombe[(vx, vy)] == 2:
                    starea_finala = 'îngeraș'
                    return starea_finala, morcovi_ronțăiți, pasi
        
        # Deplasarea iepurașului
        dx, dy = directii[directie]
        x += dx
        y += dy
        
        # Verificăm dacă toate morcovii au fost mâncați
        if all(all(cell != 7 for cell in row) for row in gradina):
            starea_finala = 'fericit'
            break
    
    return starea_finala, morcovi_ronțăiți, pasi

# Exemplu de utilizare
m = 4
n = 4
gradina = [
    [0, 7, 0, 0],
    [2, 5, 3, 0],
    [0, 6, 0, 7],
    [0, 0, 0, 0]
]
poz_initiala = (0, 0)
directie_initiala = 'est'

rezultat = simulare_gradina(m, n, gradina, poz_initiala, directie_initiala)
print(rezultat)
</syntaxhighlight>