0160 - Să se insereze într-un șir înaintea fiecărui element pătrat perfect rădăcina sa pătrată.

Sursa: InserareInainte

---

Cerinţă

Se dă un șir cu n elemente naturale. Să se insereze în șir înaintea fiecărui element pătrat perfect rădăcina sa pătrată.

Date de intrare

Programul va citi de la tastatură numărul n, apoi n numere naturale reprezentând elementele șirului.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele introduse sunt corecte!", apoi se va afișa noul vector, obținut în urma adăugării înainte fiecărui pătrat perfect, rădăcina sa pătrată. În cazul în care datele nu respectă restricțiile, se va afișa mesajul "Datele introduse nu sunt corecte!".

Restricţii şi precizări

• 1 <= n <= 25
• valorile elementelor șirului vor fi cuprinse între 0 și 1.000.000

Exemple

Exemplul 1

Intrare

Introduceti numarul de elemente: 4

Introduceti 4 elem separate prin spatiu:1 2 3 4

Ieșire

Datele introduse sunt corecte!

Noul vector este:  [1, 1, 2, 3, 2, 4]

Exemplul 2

Intrare

Introduceti numarul de elemente: 3

Introduceti 3 elem separate prin spatiu:1 245 2 4 31

Ieșire

Datele introduse sunt incorecte!

Exemplul 3

Intrare

Introduceti numarul de elemente: 5

Introduceti 5 elem separate prin spatiu:1 sfd 2 12 3

Ieșire

Datele introduse sunt incorecte!

Rezolvare

def is_integer(value):
    return value.isdigit()


def verificare_nr_elemente(n):
    if is_integer(n):
        if 0 <= int(n) <= 25:
            return n
        else:
            print("Datele introduse sunt incorecte!")
            exit()
    else:
        print("Datele introduse sunt incorecte!")
        exit()


def verificare_vector(n, vector):
    if len(vector) != int(n):
        print("Datele introduse sunt incorecte!")
        exit()
    else:
        for i in vector:
            if is_integer(i):
                if int(i) <= 1000000:
                    continue
                else:
                    print("Datele introduse sunt incorecte!")
                    exit()
            else:
                print("Datele introduse sunt incorecte!")
                exit()


def este_patrat_perfect(nr):
    if int((int(nr)) ** 0.5) ** 2 == int(nr):
        return True
    else:
        return False


def adaugare_element(vector):
    vector_nou = []
    for nr in vector:
        if este_patrat_perfect(nr):
            vector_nou.append(int((int(nr) ** 0.5)))
        vector_nou.append(int(nr))
    return vector_nou


if __name__ == '__main__':
    n = input("Introduceti numarul de elemente: ")
    verificare_nr_elemente(n)
    elem = input(f"Introduceti {n} elem separate prin spatiu:").split()
    verificare_vector(n, elem)
    print("Datele introduse sunt corecte!")
    print("Noul vector este: ", adaugare_element(elem))

Explicație

Acest cod implementează o soluție pentru problema adăugării unui element într-un vector, în funcție de paritatea elementelor existente. Utilizatorul introduce un număr de elemente și apoi elementele vectorului. Se verifică dacă datele introduse sunt corecte și, în caz afirmativ, se adaugă elemente în vector conform regulilor și se afișează noul vector.

Funcția is_integer(value) verifică dacă un caracter reprezintă un număr întreg și returnează True dacă este, altfel returnează False.

Funcția verificare_nr_elemente(n) verifică dacă numărul de elemente n este un număr întreg între 0 și 25 și returnează acest număr. Dacă n nu este valid, se afișează un mesaj de eroare și se oprește programul.

Funcția verificare_vector(n, vector) verifică dacă vectorul are n elemente și dacă fiecare element este un număr întreg mai mic sau egal cu 1.000.000. Dacă vectorul este valid, nu se întoarce nimic. Dacă nu, se afișează un mesaj de eroare și se oprește programul.

Funcția este_par(nr) verifică dacă un număr nr este par și returnează True dacă este, altfel returnează False.

Funcția adaugare_element(vector) primește un vector și adaugă un element după fiecare număr par din vector, care este dublul numărului par. Rezultatul este returnat sub formă de listă.

Instrucțiunea if __name__ == '__main__': verifică dacă programul este rulat ca script și, în caz afirmativ, apelează funcțiile verificare_nr_elemente(n) și verificare_vector(n, elem) pentru a verifica datele introduse și apoi apelează funcția adaugare_element(elem) pentru a modifica vectorul. Se afișează și un mesaj care confirmă că datele introduse sunt corecte înainte de a afișa noul vector.