0248 - Pozitie

De la Universitas MediaWiki

Sursa: 0248 - Pozitie


Cerinţa

Se dă un şir cu n numere naturale distincte două câte două. Să se determine poziţia pe care s-ar afla primul element al şirului în şirul sortat.

Date de intrare

Fişierul de intrare pozitie.in conţine pe prima linie numărul n. Urmează n numere naturale, dispuse pe mai multe linii, separate prin spaţii.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi, fişierul de ieşire pozitie.out va conţine pe prima linie numărul p, reprezentând poziţia în şirul sortat a primului element din şirul dat. În cazul în care datele nu respectă restricțiile, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse."'.

Restricţii şi precizări

  • 1 ≤ n ≤ 10000
  • elementele şirului vor avea cel mult 9 cifre
  • numerotarea elementelor se face de la 1

Exemple

Exemplul 1

pozitie.in
6
267 13 45 628 7 79
Ecran
Datele sunt introduse corect.
pozitie.out
5

Exemplul 2

pozitie.in
5
5 3 2 4
Ecran
Datele nu corespund restricțiilor impuse.


Rezolvare

# 0248

def afisare_pozitia_prim_elem_daca_sortat(vector):
    vector_sortat = sorted(vector)
    pozitie = vector_sortat.index(vector[0]) + 1
    with open('pozitie.out', 'w') as fisier_iesire:
        fisier_iesire.write(str(pozitie) + "\n")


def citire_conform_restrictiilor(n, vector):
    if n < 1 or n > 10000:
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    for element in vector:
        if element >= 1000000000:
            print("Datele nu corespund restricțiilor impuse.")
            exit()
    if n != len(vector):
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    print("Datele sunt introduse corect.")


if __name__ == '__main__':
    with open('pozitie.in', 'r') as fisier_intrare:
        n = int(fisier_intrare.readline().strip())
        vector = list(map(int, fisier_intrare.readline().strip().split()))
    citire_conform_restrictiilor(n, vector)
    afisare_pozitia_prim_elem_daca_sortat(vector)

Explicație rezolvare

   Programul de mai sus conține două funcții, funcția afisare_pozitia_prim_elem_daca_sortat(vector) și funcția citire_conform_restrictiilor(n, vector), care se vor rula în interiorul main-ului (if __name__ == '__main__' , linia 24) după citirea numărului n (linia 26) și celor n numere pe care le vom pune în șirul „vector” (linia 27), pe care le citim prin deschiderea fișierului pozitie.in (linia 25).
După ce am citit toate datele de intrare, se va apela funcția citire_conform_restrictiilor(n, vector) care primește doi parametri: n și vector. Funcția verifică n între 1 și 10000 (linia 11), dacă elementele vectorului „vector” sunt mai mici decât 1000000000 (liniile 14, 15, elementele trebuie să fie aibă cel mult 9 cifre) și dacă n este lungimea vectorului „vector” (linia 18). Dacă oricare dintre condiții este încălcată, se va afișa pe ecran mesajul „Datele nu corespund restricțiilor impuse.” și se va ieși din program cu comanda exit(). Dacă toate condițiile sunt respectate, se va afișa mesajul „Datele sunt introduse corect.” (linia 21) și se va continua programul.
Dacă s-au introdus corect datele, se va apela funcția afisare_pozitia_prim_elem_daca_sortat(vector) care primește ca parametrii numărul n și vectorul „vector”. Funcția sortează vectorul utilizând funcția implicită Python sorted() (linia 4), determină poziția primului element în șirul sortat utilizând metoda index() (tot metodă implicită Python, linia 5, 0 este poziția primului element în vectorul nostru, iar după ce aflăm indicele în vectorul sortat, îi adăugam 1 pentru a avea valoarea dacă vectorul începe de la 1), și apoi scrie rezultatul în fișierul de ieșire 'pozitie.out' utilizând funcția write().