0862 - NrSecvente

Sursa: 0862 - NrSecvente

---

Cerinţa

Se dă un vector cu n elemente, numere naturale și două numere t și k. Să se determine câte secvențe din șir au lungimea k și sunt formate din valori mai mici sau egale cu t.

Date de intrare

Programul citește de la tastatură numerele n t k, iar apoi cele n elemente ale vectorului.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi pe un rând nou va afișa pe ecran numărul C, reprezentând valoarea cerută, reprezentând valoarea cerută. În cazul contrar, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse.".

Restricţii şi precizări

• 1 ≤ k ≤ n ≤ 100.000
• t și elementele vectorului vor fi mai mici decât 1.000.000.000

Exemplu

Intrare

8 5 3

1 4 3 5 7 3 4 2

Ieșire

Datele sunt introduse corect.

3

Exemplu

Intrare

8 5 3 2

1 3 6 8 2 1 4 5

Ieșire

Datele nu corespund restricțiilor impuse.

Rezolvare

Rezolvare ver. 1

# 0862 - NrSecvente

def validate_input(n, t, k, lista):
    if not (1 <= k <= n <= 100000) or not all(1 <= x <= 1000000000 for x in [t] + arr):
        return False
    return True


def count_sequences(n, t, k, lista):
    if validate_input(n, t, k, lista):
        count = 0
        for i in range(n - k + 1):
            if all(x <= t for x in lista[i:i+k]):
                count += 1

        print("Datele sunt introduse corect.")
        print(count)
    else:
        print("Datele nu corespund restricțiilor impuse.")


if __name__ == "__main__":
    n, t, k = map(int, input().split())
    arr = list(map(int, input().split()))

    count_sequences(n, t, k, lista)

Explicatie Rezolvare

Funcția validate_input verifică dacă intrările respectă restricțiile date, iar funcția count_sequences execută codul anterior numai dacă datele de intrare sunt validate cu succes. Funcția count_sequences primește ca parametri numărul de elemente n, valoarea t, lungimea secvenței k și vectorul arr de n elemente. În primul rând se verifică dacă k este mai mare decât n sau dacă t este mai mare decât 1000000000. În aceste cazuri, se afișează un mesaj corespunzător și se returnează din funcție. Se declara o variabilă count care va fi folosită pentru a număra secvențele dorite. Se parcurge vectorul arr începând de la poziția 0 până la poziția n-k. Pentru fiecare poziție se verifică dacă toate elementele din secvența de lungime k începând de la acea poziție sunt mai mici sau egale cu t. Dacă da, se incrementează count. La final se afișează un mesaj corespunzător și valoarea lui count. În funcția main se citesc datele de intrare și se apelează funcția count_sequences.