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

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi pe un rând nou se 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, 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

15 5

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

Ieșire

Datele sunt introduse corect.

2 4

Exemplu 2

Intrare

3 2

1 1 2 3 2 6 7 8 2 4 1 1 2 3

Ieșire

Datele nu corespund restricțiilor impuse.

Rezolvare

Rezolvare ver. 1

# 0521 - kSecventa

def validate_input(n, k, arr):
    if k >= n:
        return False
    if n % k != 0:
        return False
    if max(arr) > 1000:
        return False
    return True


def find_identical_sequences(n, k, arr):
    if not validate_input(n, k, arr):
        print("Datele nu corespund restricțiilor impuse.")
        return

    seq_dict = {}
    for i in range(n // k):
        seq = tuple(arr[i * k: (i + 1) * k])
        if seq in seq_dict:
            print(seq_dict[seq], i + 1)
            return
        else:
            seq_dict[seq] = i + 1

    print("NU")


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

    if validate_input(n, k, arr):
        print("Datele sunt introduse corect.")
        find_identical_sequences(n, k, arr)
    else:
        print("Datele nu corespund restricțiilor impuse.")

Explicatie Rezolvare

Funcția validate_input(n, k, arr) verifică dacă intrarea este validă și întoarce un rezultat boolean care indică dacă intrarea este validă sau nu. Mai exact, verifică dacă k este mai mic decât n, dacă n este divizibil cu k, iar dacă toate elementele din vectorul arr sunt mai mici sau egale cu 1000.

Funcția find_identical_sequences(n, k, arr) caută și afișează pe ecran două secvențe identice de lungime k din vectorul arr, dacă există. Pentru a face acest lucru, împarte vectorul în sub-secvențe de lungime k, creând o secvență din fiecare sub-secvență și verificând dacă secvența respectivă există deja într-un dicționar. Dacă secvența există deja în dicționar, înseamnă că am găsit două secvențe identice și le afișăm pozițiile. Dacă ajungem la sfârșitul vectorului și nu găsim nicio pereche de secvențe identice, afișăm mesajul "NU".

Funcția principală if __name__ == "__main__" citește datele de intrare și, dacă acestea sunt valide, afișează mesajul "Datele sunt introduse corect." și apelează funcția find_identical_sequences(n, k, arr) pentru a căuta secvențele identice. Dacă datele de intrare nu sunt valide, se afișează mesajul "Datele nu corespund restricțiilor impuse."