4106 - Half Sort 4

Se dă un vector cu n elemente numere întregi nenule.

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 halfsort4in.txt 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 halfsort4out.txt 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

Exemplul 1

halfsort4in.txt

6

2 -1 9 -4 5 7

halfsort4out.txt

Datele introduse corespund restricțiilor impuse.

9 -4 7 -1 5 2

Exemplul 2

halfsort4in.txt

5

1 2 3 4 0

halfsort4out.txt

Datele introduse nu corespund restricțiilor impuse.

Rezolvare

# 4106 - Half Sort 4
def validare(numar_elemente, vector_intrare):           # functia de validare a datelor de intrare

    if numar_elemente > 1000 or numar_elemente < 0:
        raise ValueError

    for numar in vector_intrare:
        if abs(numar) >= 2**30 or numar == 0:    # fiecare numar trebuie sa fie mai mic decat 2^30 si nenul
            raise ValueError

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


def sorteaza_elemente(vector_intrare):                     # functia de rezolvare
    # Creăm două liste: una pentru numerele negative și una pentru cele pozitive
    neg = sorted((x for x in vector_intrare if x < 0), reverse=True)
    pos = sorted((x for x in vector_intrare 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 vector_intrare]


if __name__ == '__main__':
    file_in = open("halfsort4in.txt", "r")         # declararea fisierelor
    file_out = open("halfsort4out.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    # din cauza datelor de intrare pot aparea 2 tipuri de erori, valueError sau IndexError pe care le tratam
    try:
        n = int(file_in.readline())      # citirea numarului de elemente
        vector = list(map(int, file_in.readline().split()))  # citirea vectorului

        validare(n, vector)                 # apelul functiei de validare
        vector = sorteaza_elemente(vector)               # apelul functiei de rezolvare
        file_out.write(' '.join(map(str, vector)) + '\n')

    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")