Editing 2102 - Robot 3 (section)

=== Exemplul 3: ===
<code>robot3.in</code>
 3
 19332
<code>robot3.out</code>
 D1AS2AD4AAS1AT
<code>C=3</code>, pentru acest test se rezolvă <code>cerința 3</code>. Pentru a afișa cifra <code>1</code>, mânerul robotului se deplasează <code>1</code> unitate la dreapta după care apasă <code>(D1A)</code>. Pentru a afișa cifra <code>9</code>, din poziția curentă mânerul robotului se deplasează <code>2</code> unități la stânga şi apasă <code>(S2A)</code>. Pentru a afișa cifra <code>3</code>, din poziția curentă mânerul robotului se deplasează <code>4</code> unități la dreapta după care apasă <code>(D4A)</code>. Pentru a afișa a doua cifra <code>3</code>, mânerul robotului rămâne în poziția curentă și apasă butonul. Pentru a afișa cifra <code>2</code>, din poziția curentă mânerul robotului se deplasează <code>1</code> unitate la stânga după care apasă <code>(S1A)</code>. Programul se termină cu instrucțiunea <code>T → D1AS2AD4AAS1AT</code><syntaxhighlight lang="python3">
def main():
    # Deschidem fisierul de intrare si iesire
    with open("robot.in", "r") as fin, open("robot.out", "w") as fout:
        # Citim numarul de cazuri de test 'numar_c'.
        numar_c = int(fin.readline().strip())

        # Citim sirul 'numar_a' de caractere A, S, D.
        numar_a = fin.readline().strip()

        # Definim functia 'rezultat' care va scriem rezultatul in fisier in functie de diferenta dintre doua caractere.
        def rezultat(a, b):
            if a == b:
                fout.write('A')
            elif a < b:
                if b - a <= (a + 10 - b) % 10:
                    fout.write(f'D{b - a}A')
                elif b - a > (a + 10 - b) % 10:
                    fout.write(f'S{(a + 10 - b) % 10}A')
            else:
                if a - b < (b + 10 - a) % 10:
                    fout.write(f'S{a - b}A')
                else:
                    fout.write(f'D{(b + 10 - a) % 10}A')

        # Verificam tipul de operatie specificat de 'numar_c'.
        if numar_c == 1:
            # Calculam numarul de A-uri si il scriem in fisier.
            numar = numar_a.count('A')
            fout.write(str(numar))
        elif numar_c == 2:
            # Calculam pozitia finala in functie de operatiile S si D si o scriem in fisier.
            pozitie = 0
            for i in range(len(numar_a)):
                if numar_a[i] == 'S':
                    variabila = int(numar_a[i + 1])
                    pozitie = (pozitie + 10 - variabila) % 10
                elif numar_a[i] == 'D':
                    variabila = int(numar_a[i + 1])
                    pozitie = (pozitie + variabila) % 10
                elif numar_a[i] == 'A':
                    fout.write(str(pozitie))
        elif numar_c == 3:
            # Calculam rezultatul pentru fiecare pereche de caractere si scriem 'T' la final.
            rezultat(0, int(numar_a[0]))
            for i in range(1, len(numar_a)):
                rezultat(int(numar_a[i - 1]), int(numar_a[i]))
            fout.write('T')

if __name__ == "__main__":
    main()
</syntaxhighlight>