0522 - Excursie

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 ≤ n ≤ 1000
• 1 ≤ k ≤ 15
• cele n numere citite vor fi mai mici decât 1000
• elementele vectorului sunt indexate de la 1 la n

Exemplul 1

Intrare

12 5

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

Ieșire

Datele de intrare corespund restricțiilor impuse.

2 6

Exemplul 2

Intrare

0

0

Ieșire

Datele de intrare nu corespund restricțiilor impuse.

Rezolvare

#0522 - Excursie
def secvente_identice(n, k, elemente):
    gasit = False
    for i in range(1, n - k + 2):  # parcurgem sirul de la pozitia 1 pana la pozitia n-k+1
        for j in range(i + k - 1, n - k + 2):  # parcurgem sirul de la pozitia i+k-1 pana la pozitia n-k+1
            if elemente[i:i+k] == elemente[j:j+k]:  # verificam daca secventele de lungime k de la pozitia i si j sunt identice
                gasit = True
                print(i, j)  # afisam pozitiile in care am gasit secventele identice
                break  # iesim din a doua bucla
        if gasit:
            break  # iesim din prima bucla, daca am gasit o secventa identica
    if not gasit:
        print("NU")
if __name__ == '__main__':
    # citirea datelor de intrare
    n = int(input())  # citim numarul de elemente din sir
    k = int(input())  # citim lungimea secventei cautate
    elemente = [0] + [int(x) for x in input().split()]  # citim elementele sirului, adaugand 0 la inceputul listei pentru a nu afecta pozitia indexului
    
    # verificarea restricțiilor impuse
    if n >= 1 and n <= 1000 and k >= 1 and k <= 15:
        secvente_identice(n, k, elemente)  # apelam functia care va cauta secventele identice
        print("Datele de intrare corespund restricțiilor impuse.")
    else:
        print("Datele de intrare nu corespund restricțiilor impuse.")