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

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi pe un rând nou 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, 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 ≤ 1000, k este divizor al lui n
cele n numere citite vor fi mai mici decât 1000

Exemplu 1

Intrare: 12 5; 2 3 1 1 4 3 1 1 4 3 8 8
Ieșire: Datele sunt introduse corect.; 2 6

Exemplu 2

Intrare: 12 5; 2 3 1 1 4 3 1 1 4 3 8 8
Ieșire: Datele nu corespund restricțiilor impuse.; 3 5

Rezolvare

Rezolvare ver. 1

0522 - kSecventa1

def kSecventa(n, k, v):

   if n % k != 0:
       print("Datele nu corespund restricțiilor impuse.")
       return

   # Impartim lista in k bucati cat mai egale
   chunk_size = n // k
   extra = n % k
   chunks = []
   start = 0
   for i in range(k):
       end = start + chunk_size
       if extra > 0:
           end += 1
           extra -= 1
       chunks.append(v[start:end])
       start = end

   # Verificam daca exista doua secvente identice
   for i in range(k):
       for j in range(i+1, k):
           if chunks[i] == chunks[j]:
               print("Datele sunt introduse corect.")
               print(i*chunk_size+1, j*chunk_size+1)
               return

   print("Datele sunt introduse corect.")
   print("NU")

if __name__ == "__main__":

   n, k = map(int, input().split())
   if not (isinstance(n, int) and isinstance(k, int) and n > 1 and k > 1):
       print("Datele nu corespund restricțiilor impuse.")
       exit()

   if n % k != 0:
       print("Datele nu corespund restricțiilor impuse.")
       exit()

   v = list(map(int, input().split()))
   if not (len(v) == n and all(isinstance(x, int) and x > 0 and x < 1000 for x in v)):
       print("Datele nu corespund restricțiilor impuse.")
       exit()

   kSecventa(n, k, v)

</syntaxhighlight>

Explicatie Rezolvare

Funcția kSecventa primește ca argumente lungimea vectorului n, lungimea secvențelor k și vectorul v. În cazul în care lungimea vectorului nu se împarte exact la lungimea secvențelor, se afișează mesajul "Datele nu corespund restricțiilor impuse." și se încheie execuția funcției. Altfel, se împarte vectorul în k bucăți cât mai egale și se verifică dacă există două secvențe identice. Dacă se găsește o astfel de secvență, se afișează mesajul "Datele sunt introduse corect." și pozițiile de început ale celor două secvențe. În caz contrar, se afișează mesajul "Datele sunt introduse corect." și "NU". In funcția if __name__ == "__main__":, valorile de intrare sunt citite și apoi apelăm funcția kSecventa(n, k, v) cu valorile citite ca argumente. În plus, validarea datelor de intrare este realizată în același mod ca în codul anterior. Deci, nu trebuie să rescriem codul pentru funcția kSecventa(n, k, v) în funcția if __name__ == "__main__":..