4106 - Half Sort 4

Cerinţa

Să se ordoneze crescător subșirul format din elementele negative din vector și descrescător subșirul format din elementele pozitive. Indicii subșirului format din elementele negative nu se vor schimba. La fel și indicii subșirului format din elementele pozitive.

Date de intrare

Fişierul de intrare halfsort4.in conţine pe prima linie numărul n si pe a doua linie n numere întregi nenule separate prin spaţii.

Date de ieşire

Fişierul de ieşire halfsort4.out va conţine pe prima linie cele n elemente ale vectorului, ordonate conform cerinței, separate printr-un spațiu.

Restricții și precizări

• 0 < n ≤ 1000
• valoarea absolută a numerelor de pe a doua linie a fişierului de intrare va fi mai mică decât 2^30

Exemplu

halfsort4.in

6

2 -1 9 -4 5 7

halfsort4.out

9 -4 7 -1 5 2

Rezolvare

<syntaxhighlight lang="python" line> def halfsort(n, arr):

   # Creăm două liste: una pentru numerele negative și una pentru cele pozitive
   neg = sorted((x for x in arr if x < 0), reverse=True)
   pos = sorted((x for x in arr if x > 0))

   # Reconstruim lista inițială. Pentru fiecare element din lista inițială:
   # - dacă este negativ, adăugăm cel mai mic număr negativ rămas (și îl eliminăm din listă)
   # - dacă este pozitiv, adăugăm cel mai mare număr pozitiv rămas (și îl eliminăm din listă)
   return [neg.pop() if x < 0 else pos.pop() for x in arr]

if __name__ == "__main__":

   # Această parte a codului va fi executată doar dacă scriptul este rulat direct

   # Testăm funcția cu datele din exemplu
   n = 6
   arr = [2, -1, 9, -4, 5, 7]

   # Afișăm rezultatul
   print(halfsort(n, arr))

</syntaxhighlight>