Editing 2102 - Robot 3

Paul dorește să învețe cum să programeze un robot. Pentru început s-a gândit să construiască un robot format dintr-un mâner, <code>10</code> butoane aranjate circular şi un ecran. Pe butoane sunt scrise, în ordine crescătoare, cifrele de la <code>0</code> la <code>9</code>, ca în figură.
[[Fișier:Image.png|stanga|miniatura]]
Un roboprogram va fi format dintr-o secvenţă de instrucţiuni. Instrucțiunile pot fi:

Iniţial mânerul robotului este plasat în dreptul butonului <code>0</code>, iar ecranul este gol.

De exemplu, în urma executării roboprogramului <code>D4AS1AAD6AT</code> robotul apasă butoanele pe care sunt scrise cifrele <code>4, 3, 3, 9,</code> iar pe ecran va apărea <code>4339</code>.

=== Cerințe ===
Să se scrie un program care rezolvă următoarele cerinţe:

=== Date de intrare ===
Fişierul de intrare <code>robot3.in</code> conţine pe prima linie un număr natural <code>C</code>, reprezentând cerinţa care urmează să fie rezolvată <code>(1, 2 sau 3)</code>. Dacă <code>C=1</code> sau <code>C=2</code>, pe a doua linie a fişierului se află un roboprogram. Dacă <code>C=3</code>, pe a doua linie a fişierului de intrare se află numărul natural <code>N</code>.

=== Date de ieșire ===
Fişierul de ieşire <code>robot3.out</code> va conţine o singură linie. Dacă <code>C=1</code>, pe prima linie se va scrie un număr natural reprezentând numărul de cifre afişate pe ecran după executarea roboprogramului din fişierul de intrare.

Dacă <code>C=2</code>, pe prima linie vor fi scrise cifrele afișate pe ecran în urma executării roboprogramului din fişierul de intrare. Dacă <code>C=3</code>, pe prima linie va fi scris roboprogramul solicitat de cerinţa <code>3</code>.

=== Restricții și precizări ===
• <code>0 ≤ N ≤ 1000000000</code>

• Lungimea roboprogramului citit din fişierul de intrare sau scris în fişierul de ieşire este <code>cel mult 1000</code> de caractere.

• Dacă mânerul este plasat în dreptul butonului <code>0</code> şi se deplasează spre dreapta, se va îndrepta către butonul <code>1</code>; dacă deplasarea este spre stânga, se va îndrepta către butonul <code>9</code>.

• Pentru rezolvarea corectă a primei cerinţe se acordă 10 puncte, pentru rezolvarea corectă a celei de a doua cerințe se acordă 30 de puncte, iar pentru rezolvarea corectă a celei de a treia cerințe se acordă 50 de puncte. 10 puncte se acordă din oficiu.

=== Exemplul 1: ===
<code>robot3.in</code>
 1
 D1AD2AS1AT
<code>robot3.out</code>
 3
<code>C=1</code>, pentru acest test se rezolvă <code>cerința 1</code>.

Se afişează pe ecran <code>3 cifre (132)</code>

=== Exemplul 2: ===
<code>robot3.in</code>
 2
 S0AD2AS1AT
<code>robot3.out</code>
 021
<code>C=2</code>, pentru acest test se rezolvă <code>cerința 2</code>.

Mânerul robotului se deplasează cu <code>0</code> unități la stânga, deci rămâne în dreptul butonului <code>0</code> și apasă, apoi se deplasează <code>2</code> unități spre dreapta şi ajunge în dreptul butonului <code>2</code>, apasă, apoi se deplasează <code>1</code> unitate la stânga și ajunge în dreptul butonului <code>1</code> și apasă acest buton → <code>021</code>.

=== 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>