Editing 1031 - Culori2 (section)

= Exemplu 2: =
<code>culori2in.txt</code>
 0
<code>culori2out.txt</code>
 Datele de intrare nu au fost respectate.
== Rezolvare ==
<syntaxhighlight lang="python">
def numar_moduri_vopsire_gard(N):
    # Inițializăm un dicționar pentru a stoca numărul de moduri posibile
    # pentru fiecare culoare și fiecare poziție
    moduri = {
        'alb': [0] * (N + 1),
        'albastru': [0] * (N + 1),
        'rosu': [0] * (N + 1),
        'verde': [0] * (N + 1),
        'galben': [0] * (N + 1)
    }

    # Pentru prima scândură, fiecare culoare este o opțiune validă
    for culoare in moduri.keys():
        moduri[culoare][1] = 1

    # Calculăm numărul de moduri pentru fiecare poziție și culoare
    for i in range(2, N + 1):
        moduri['alb'][i] = moduri['albastru'][i - 1]
        moduri['albastru'][i] = moduri['alb'][i - 1] + moduri['rosu'][i - 1]
        moduri['rosu'][i] = moduri['albastru'][i - 1] + moduri['verde'][i - 1]
        moduri['verde'][i] = moduri['rosu'][i - 1] + moduri['galben'][i - 1]
        moduri['galben'][i] = moduri['verde'][i - 1]

    # Sumăm modurile pentru toate culorile pentru a obține rezultatul final
    rezultat = sum(moduri[culoare][N] for culoare in moduri.keys())
    return rezultat


def main():
    try:
        with open("culori2in.txt", "r") as file_in:
            N = int(file_in.readline().strip())

            if 1 <= N <= 5000:
                rezultat = numar_moduri_vopsire_gard(N)

                with open("culori2out.txt", "w") as file_out:
                    file_out.write(str(rezultat))
            else:
                print("Datele de intrare nu au fost respectate.")
    except Exception as e:
        print(f"Nu au fost respectate cerințele impuse: {e}")


if __name__ == "__main__":
    main()

</syntaxhighlight>