Editing 3658 – NrTriplete

Sursa: [https://www.pbinfo.ro/probleme/3658/nrtriplete NrTriplete]
----
==Cerinţă==
Se dă un șir cu <code>n</code> elemente întregi și s, un nr întreg. Funcția returnează numărul de triplete <code>(sir[i], sir[j], sir[k])</code>, unde <code>0 ≤ i < j < k < n</code> și sir<code>[i] + sir[j] + sir[k] < S</code>.
==Date de intrare==
Programul va citi de la tastatură valoarea<code>n</code>, apoi <code>n</code> numere întregi reprezentând elementele șirului, apoi valoarea s.
==Date de ieșire==
Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele introduse sunt corecte!", apoi se va afișa totalul de triplete care îndeplinesc condițiile date. În cazul în care datele nu respectă restricțiile, se va afișa mesajul "Datele introduse nu sunt corecte!".
==Restricţii şi precizări==
*3 <= n <= 2000
*valorile elementelor șirului vor fi  cuprinse între -100.000.000 și 100.000.000
*-100.000.000 <= s <= 100.000.000
==Exemple==
===Exemplul 1===
;Intrare
:Introduceti numarul de elemente a sirului: 4
:Introduceti 4 numere separate prin spatiu:1 -2 0 4
:Introduceti valoarea s: 5
;Ieșire
:Datele introduse sunt corecte!
:count: 3
===Exemplul 2===
;Intrare
:Introduceti numarul de elemente a sirului: 3
:Introduceti 3 numere separate prin spatiu: 1231 1 321 41
;Ieșire
:Datele introduse sunt incorecte!
==Rezolvare==
<syntaxhighlight lang="python" line="1">
def is_integer(value):
    return value.isdigit()


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


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


def verificare_s(s):
    if is_integer(s):
        if 3 <= int(s) <= 2000:
            return s
        else:
            print("Datele introduse sunt incorecte!")
            exit()
    else:
        print("Datele introduse sunt incorecte!")
        exit()


def nr_triplete(sir, n, S):
    count = 0
    for i in range(n-2):
        for j in range(i+1, n-1):
            for k in range(j+1, n):
                if sir[i]+sir[j]+sir[k] < S:
                    count += 1
    print("count:",count)


if __name__ == '__main__':
    n = input("Introduceti numarul de elemente a sirului: ")
    verificare_nr_elemente(n)
    lst_int = list(map(int, input(f"Introduceti {n} numere separate prin spatiu:").split()))
    verificare_vector(n, lst_int)
    s = input("Introduceti valoarea s: ")
    verificare_s(s)
    print("Datele introduse sunt corecte!")
    nr_triplete(lst_int, int(n), int(s))




</syntaxhighlight>
==Explicație rezolvare==

# Funcția <code>is_integer(value)</code> verifică dacă <code>value</code> este un string reprezentând un număr întreg (adică conține doar cifre). Returnează <code>True</code> dacă este și <code>False</code> în caz contrar.
# Funcția <code>verificare_nr_elemente(n)</code> verifică dacă <code>n</code> este un string reprezentând un număr întreg între 3 și 2000. În cazul în care este valid, returnează <code>n</code>, altfel afișează un mesaj de eroare și iese din program.
# Funcția <code>verificare_vector(n, vector)</code> verifică dacă toate elementele din <code>vector</code> sunt numere întregi între -100000000 și 100000000. În cazul în care sunt valide, nu face nimic, altfel afișează un mesaj de eroare și iese din program.
# Funcția <code>verificare_s(s)</code> verifică dacă <code>s</code> este un string reprezentând un număr întreg între 3 și 2000. În cazul în care este valid, returnează <code>s</code>, altfel afișează un mesaj de eroare și iese din program.
# Funcția <code>nr_triplete(sir, n, S)</code> primește un tablou <code>sir</code> de <code>n</code> numere întregi și un număr întreg <code>S</code>. Parcurge toate tripletele posibile din tablou și numără cele pentru care suma este strict mai mică decât <code>S</code>. Afișează rezultatul final.
# În funcția <code>main</code>, se citesc numărul de elemente, vectorul și valoarea <code>S</code> cu ajutorul unor input-uri și se validează folosind funcțiile de mai sus. Apoi, se apelează funcția <code>nr_triplete</code> cu argumentele corespunzătoare.
# Comanda <code>list(map(int, input(f"Introduceti {n} numere separate prin spatiu:").split()))</code> transformă string-ul citit de la tastatură într-o listă de numere întregi. Mai întâi se citesc valorile cu <code>input(f"Introduceti {n} numere separate prin spatiu:")</code>, apoi se separă folosind funcția <code>split()</code> și se transformă fiecare element din listă într-un număr întreg folosind funcția <code>int</code>.