0522 - kSecventa1

Sursa: 0522 - kSecventa1

---

Cerinţa

Se dă un vector cu n elemente, numere naturale, și un număr k. Să se stabilească dacă în vector există două secvențe de lungime k identice.

Date de intrare

Programul citește de la tastatură numerele n și k, iar apoi n numere naturale, reprezentând elementele vectorului.

Date de ieșire

Programul va afișa pe ecran numerele i j, i < j reprezentând pozițiile de început a două secvențe de lungime k identice, dacă există două astfel de secvențe, sau mesajul NU, dacă nu există două astfel de secvențe. Dacă există mai multe perechi de secvențe identice se vor considera cele cu numerele de ordine i j minime.

Restricţii şi precizări

• 1 ≤ k < n ≤ 1000, k este divizor al lui n
• cele n numere citite vor fi mai mici decât 1000

Exemplu

Intrare

12 5

2 3 1 1 4 3 1 1 4 3 8 8

Ieșire

2 6

Rezolvare

Rezolvare ver. 1

# 0522 - kSecventa1

def citire_date():
    n = int(input())
    k = int(input())
    v = list(map(int, input().split()))
    return n, k, v

def gaseste_secvente_identice(n, k, v):
    secvente = {}
    for i in range(n - k + 1):
        secventa = tuple(v[i:i+k])
        if secventa in secvente:
            return secvente[secventa], i
        else:
            secvente[secventa] = i
    return "NU"

def valideaza(n, k, v):
    if n <= k:
        return "NU"
    return " ".join(str(i) for i in gaseste_secvente_identice(n, k, v))

if __name__ == "__main__":
    n, k, v = citire_date()
    print(valideaza(n, k, v))

Explicatie Rezolvare

Funcția citire_date() primește input-ul și returnează cele trei variabile citite. Funcția gaseste_secvente_identice(n, k, v) primește dimensiunea vectorului n, lungimea secvențelor k și vectorul v, și returnează o pereche de numere care reprezintă pozițiile de început ale celor două secvențe identice sau un string "NU" în caz contrar. Pentru a găsi secvențele identice, se parcurg toate secvențele de lungime k din vector, iar pentru fiecare se construiește un tuplu și se verifică dacă apare într-un dicționar de secvențe întâlnite anterior. Dacă apare, se returnează pozițiile celor două secvențe identice, altfel se adaugă tuplul în dicționar. Funcția valideaza(n, k, v) primește aceleași trei variabile și returnează "NU" dacă n<=k sau rezultatul funcției gaseste_secvente_identice(n, k, v) altfel. Dacă există două secvențe identice, se întorc pozițiile lor, altfel se întoarce string-ul "NU". Blocul if __name__ == "__main__": este necesar pentru a permite rularea funcțiilor din consolă. În acest bloc se citesc datele de intrare, se validează și se afișează rezultatul.