Editing 2650 - books (section)

==Rezolvare==

<syntaxhighlight lang="python" line="1" start="1">

def verificare_restrictii(nr_n):
    # Functia de verificare a datelor de intrare
    if 1 <= nr_n <= 200000:
        return True
    else:
        return False


def muta_carti(nr_n, nr_a, nr_b):
    # Verificam daca numerele sunt distincte
    if len(set(a)) != nr_n or len(set(nr_b)) != nr_n:
        return "Numerele nu sunt distincte între ele."

    # Creăm un array poz cu lungimea n + 1
    poz = [0] * (nr_n + 1)

    # Parcurgem array-ul a și pentru fiecare element a[i], setăm poz[a[i]] la i + 1
    for i in range(nr_n):
        poz[nr_a[i]] = i + 1

    # Inițializăm rezultatul ca un array gol
    rez = []

    # Inițializăm max_poz la 0
    max_poz = 0

    # Parcurgem array-ul b
    for i in range(nr_n):
        # Dacă poz[b[i]] este mai mare decât max_poz, adăugăm diferența la rezultat și actualizăm max_poz
        if poz[nr_b[i]] > max_poz:
            rez.append(poz[b[i]] - max_poz)
            max_poz = poz[b[i]]
        else:
            # Altfel, adăugăm 0 la rezultat
            rez.append(0)

    # Returnăm rezultatul
    return rez


if __name__ == "__main__":
    try:
        # Introdu datele de intrare manual
        n = int(input("Introdu numărul de cărți: "))

        # Verificam daca datele de intrare corespund restrictiilor
        if verificare_restrictii(n):
            print("Datele de intrare corespund restrictiilor impuse.")

            # Convertim string-ul de intrare într-un array de numere întregi
            a = list(map(int, input("Introdu ordinea cărților în stack: ").split()))

            # Convertim string-ul de intrare într-un array de numere întregi
            b = list(map(int, input("Introdu ordinea în care vrei să muți cărțile: ").split()))

            # Afișează rezultatul
            print(muta_carti(n, a, b))
        else:
            print("Datele de intrare nu corespund restrictiilor impuse.")
    except ValueError:
        print("Datele de intrare nu corespund restrictiilor impuse.")
    except IndexError:
        print("Datele de intrare nu corespund restrictiilor impuse.")

</syntaxhighlight>