Editing 3086 - densitate1 (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
#3086 - Densitate1

def validare_date(n, a, d):
    # Verificăm restricțiile privind lungimea șirului
    if not 1 <= n <= 100000:
        return False, "Lungimea șirului trebuie să fie între 1 și 100.000."

    # Verificăm restricțiile privind elementele șirului
    for elem in a:
        if not 0 <= elem <= 1000000:
            return False, "Elementele șirului trebuie să fie între 0 și 1.000.000."

    # Verificăm restricțiile privind densitatea d
    if not 0 <= d <= 0.99:
        return False, "densitatea trebuie să fie între 0 și 0.99."

    return True, ""


def calculeaza_secvente(n, a, d):
    numar_secvente = 0

    # Parcurgem șirul pentru a genera secvențele
    for i in range(n):
        for j in range(i + 1, n + 1):
            # Extragem secvența curentă
            secventa_curenta = a[i:j]

            # Calculăm densitatea secvenței curente
            densitate_secventa = len([x for x in secventa_curenta if x % 2 == 0]) / len(secventa_curenta)

            # Verificăm dacă densitatea este exact d
            if round(densitate_secventa, 2) == d:
                numar_secvente += 1

    return numar_secvente


# Citim datele din fișierul de intrare
with open("densitatein.txt", "r") as f:
    n = int(f.readline())
    a = list(map(int, f.readline().split()))
    d = float(f.readline())

# Validăm datele
valid, mesaj = validare_date(n, a, d)

# dacă datele sunt valide, rezolvăm problema și scriem rezultatul în fișierul de ieșire
if valid:
    print("Datele de intrare corespund restricțiilor impuse")
    rezultat = calculeaza_secvente(n, a, d)
    with open("densitateout.txt", "w") as f_out:
        f_out.write(str(rezultat))
else:
    print("Datele de intrare NU corespund restricțiilor impuse")

</syntaxhighlight>