0521 - kSecventa

Sursa: 0521 - kSecventa

---

Cerinţa

Se dă un vector cu n elemente, numere naturale, și un număr k, divizor al lui n. Se împarte vectorul în k secvențe disjuncte, numerotate de la 1 la k. Să se stabilească dacă există două secvențe 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 numerele de ordine a două secvențe 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 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

15 5

3 6 4 7 6 6 6 4 6 7 6 6 4 4 5

Ieșire

2 4

Rezolvare

Rezolvare ver. 1

# 0521 - kSecventa

def citeste_date_intrare():
    n, k = map(int, input().split())
    x = list(map(int, input().split()))
    return n, k, x

def gaseste_secvente_identice(n, k, x):
    l = n // k
    seq_dict = {}
    for i in range(1, k+1):
        seq = tuple(x[(i-1)*l:i*l])
        if seq not in seq_dict:
            seq_dict[seq] = 1
        else:
            seq_dict[seq] += 1
        if seq_dict[seq] == 2:
            for j in range(1, i):
                if seq == tuple(x[(j-1)*l:j*l]):
                    return j, i
    return "NU"

def valideaza_datele_de_intrare(n, k, x):
    assert 1 <= k < n <= 1000 and n % k == 0, "Datele de intrare sunt incorecte!"
    assert max(x) < 1000, "Datele de intrare sunt incorecte!"

if __name__ == '__main__':
    n, k, x = citeste_date_intrare()
    valideaza_datele_de_intrare(n, k, x)
    rezultat = gaseste_secvente_identice(n, k, x)
    print(rezultat)

Explicatie Rezolvare

Citim datele de intrare (n, k și vectorul x). Calculăm lungimea fiecărei secvențe (l), care este egală cu n/k. Inițializăm un dicționar (seq_dict) în care vom stoca fiecare secvență și numărul de apariții a acesteia. Iterăm prin fiecare secvență (de la 1 la k) și: a. Extragem secvența corespunzătoare. b. Dacă secvența nu există deja în dicționar, o adăugăm cu valoarea 1. c. Dacă secvența există deja în dicționar, o incrementăm cu 1. d. Verificăm dacă secvența a fost întâlnită de două ori și, dacă da, afișăm index-urile secvențelor identice și ieșim din program. Dacă nu am găsit două secvențe identice, afișăm "NU".