Editing 2279 - Concurs Perf

==Cerința==
Chef vrea să organizeze un concurs. Previziunea nivelurilor de dificultate a problemelor poate fi o sarcină descurajantă. Chef vrea să echilibreze nivelurile de dificultate ale problemelor.

Să presupunem că un concurs a avut <span style=“color: red”> P</span> participanți în total. <span style=“color: red”> O</span> problemă care a fost rezolvată de cel puțin jumătate dintre participanți (adică <span style=“color: red”> P/2</span>) este considerată ușoară. <span style=“color: red”> O</span> problemă rezolvată de participanții de cel mult <span style=“color: red”> P/10</span> este considerată dificilă.


Chef vrea echilibru. Potrivit lui, un concurs echilibrat trebuie să aibă exact <span style=“color: red”> 1</span> problemă ușoară și exact <span style=“color: red”> 2</span> probleme grele. Vi se oferă descrierea problemelor <span style=“color: red”> N</span> și numărul de participanți care rezolvă aceste probleme. Puteți spune dacă concursul a fost echilibrat sau nu?

==Date de intrare==
Primul rând conține un număr întreg <span style=“color: red”> T</span> care indică numărul de cazuri de testare.
Prima linie a fiecărui caz de testat conține două numere întregi <span style=“color: red”> N</span> și <span style=“color: red”> P</span> indicând numărul de probleme, respectiv numărul de participanți.
Cea de-a doua linie conține <span style=“color: red”> N</span> numere întregi, separate prin spațiu, al căror număr indică numărul de participanți care rezolvă problema cu numărul de ordine <span style=“color: red”> i</span>.

==Date de ieșire==
Pentru fiecare caz, afișati da, dacă concursul este echilibrat, sau nu.

==Restricții și precizări==
<span style=“color: red”> 1 ≤ T, N ≤ 500
1 ≤ P ≤ 10^8^
1 ≤ numărul de participanți care rezolvă o problemă ≤ P</span>

==Exemplu:==
Intrare
6
3 100
10 1 100
3 100
11 1 100
3 100
10 1 10
3 100
10 1 50
4 100
50 50 50 50
4 100
1 1 1 1

Ieșire
da
nu
nu
da
nu
nu

==Rezolvare==
<syntaxhighlight lang="python" line>
def validate(t, n, p, problems):
    if t <= 0 or n <= 0 or p <= 0:
        return False
    if len(problems) != n:
        return False
    if not all(1 <= x <= p for x in problems):
        return False
    return True

def read_input():
    t = int(input())
    test_cases = []
    for i in range(t):
        n, p = map(int, input().split())
        problems = list(map(int, input().split()))
        if not validate(t, n, p, problems):
            return None
        test_cases.append((n, p, problems))
    return test_cases

def is_balanced(n, p, problems):
    easy = 0
    hard = 0
    for i in range(n):
        if problems[i] > p // 2:
            easy += 1
        elif problems[i] <= p // 10:
            hard += 1
    return easy == 1 and hard == 2

def main():
    test_cases = read_input()
    if test_cases is None:
        print("Invalid input")
        return
    for case in test_cases:
        n, p, problems = case
        if is_balanced(n, p, problems):
            print("da")
        else:
            print("nu")

if __name__ == '__main__':
    main()
</syntaxhighlight>

==Explicatie==
În funcția validate, verificăm dacă datele de intrare sunt valide, adică dacă numerele sunt în intervalul specificat și dacă lista problemelor are dimensiunea corectă.

În funcția read_input, citim datele de intrare pentru fiecare caz de testare și le validăm folosind funcția validate.

În funcția is_balanced, verificăm dacă un anumit caz de testare respectă condițiile problemei, adică dacă există o singură problemă ușoară și două probleme grele.

În funcția main, citim datele de intrare pentru fiecare caz de testare și le procesăm folosind funcția is_balanced.