0556 - Flici

De la Universitas MediaWiki

Cerinţa

Un flic este o creatură pufoasă de dimensiunea unui hamster, având trei ochi și o blană colorată. De la naștere, fiecărui flic îi place în mod deosebit un anumit număr.

Hobby-ul lor este să intre în cutii, iar în lumea flicilor, pe fiecare cutie este inscripționat un număr. Flicii sunt pretențioși și nu vor alege orice cutie. În mod ideal, ar alege cutia pentru care numărul inscripționat este cel mai apropiat de numărul lor favorit, dar pentru că flicii sunt altruiști, vor alege cutiile astfel încât ceilalți flici să nu se supere prea tare.

Astăzi s-a format un grup de n flici, fiecare cu un număr favorit, care au la dispoziție n cutii, fiecare având inscripționat un număr. Sarcina ta este să stabilești pentru fiecare flic în ce cutie va intra, astfel încât suma modulelor diferențelor dintre numărul favorit a flicului și cel inscripționat pe cutia în care intră acesta să fie minimă.

Date de intrare

Fișierul de intrare fliciin.txt conține pe prima linie numărul n; a doua linie conține n numere naturale separate prin spații, reprezentând numerele favorite ale flicilor, iar a treia linie conține n numere naturale separate prin spații, reprezentând numerele inscripționate pe cutii.

Date de ieșire

Fișierul de ieșire fliciout.txt va conține pe prima linie numărul S, reprezentând suma minimă obținută.

Restricţii şi precizări

  • 1 ⩽ n ⩽ 100.000
  • numerele favorite și cele inscripționate sunt numere naturale mai mici sau egale cu 10.000
  • pentru 40% din punctaj, n < 10
  • pentru 80% din punctaj, n ⩽ 1.000

Exemplu 1

fliciin.txt
3
1 1 3
1 2 2
fliciout.txt
Datele de intrare corespund restrictiilor impuse
2


Exemplu 2

fliciin.txt
3
1 1 3000000
1 2 2
fliciout.txt
Datele de intrare nu corespund restrictiilor impuse


Explicatie

Flicul 1 va intra în cutia 1, flicul 2 în a doua, flicul 3 în a treia, rezultând suma |1-1| + |1-2| + |3-2| = 2.

Rezolvare

def suma_minima(n, flici, cutii):
    # Calculează suma minimă a diferențelor dintre numărul favorit al flicului și cel inscripționat pe cutie.

    flici.sort()
    cutii.sort()
    suma = 0

    for i in range(n):
        suma += abs(flici[i] - cutii[i])

    return suma


def main():
    with open('fliciin.txt', 'r') as fin, open('fliciout.txt', 'w') as fout:
        n = int(fin.readline().strip())
        flici = list(map(int, fin.readline().split()))
        cutii = list(map(int, fin.readline().split()))

        # Verifică dacă datele de intrare respectă restricțiile
        if not (1 <= n <= 100000 and all(1 <= x <= 10000 for x in flici + cutii)):
            fout.write("Datele de intrare nu corespund restrictiilor impuse\n")
            return

        fout.write("Datele de intrare corespund restrictiilor impuse\n")

        suma = suma_minima(n, flici, cutii)
        fout.write(f"{suma}\n")


if __name__ == "__main__":
    main()