0556 - Flici

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

<syntaxhighlight lang="python" line> 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()

</syntaxhighlight>