Editing 1610 - Colier (section)

== Rezolvare == 
<syntaxhighlight lang="python" line>
#1610
def solve(t, n, margele):
    tip1 = []
    tip2 = []
    for m in margele:
        if m % 10 > m // 10:
            tip2.append(m)
        else:
            tip1.append(m)

    if t == 1:
        return len(tip1)
    elif t == 2:
        num_margele = len(margele)
        max_margele = 0
        for i in range(num_margele):
            num = 1
            last_tip = 1 if margele[i] in tip1 else 2
            for j in range(i + 1, num_margele + i):
                index = j % num_margele
                if margele[index] in tip1:
                    tip = 1
                else:
                    tip = 2
                if tip != last_tip:
                    num += 1
                    last_tip = tip
            if num > max_margele:
                max_margele = num

        return max_margele


def validare_input(t, n, margele):
    # verificăm restricțiile pentru T
    if t not in [1, 2]:
        return False
    # verificăm restricțiile pentru N
    if n < 1 or n > 50000:
        return False
    # verificăm restricțiile pentru mărgele
    for m in margele:
        if m < 12 or m > 987654321:
            return False
        digits = set(str(m))
        if len(digits) != len(str(m)) or '0' in digits:
            return False
    return True


if __name__ == '__main__':
    t = int(input("T: "))
    n = int(input("N: "))
    margele = list(map(int, input("Margele: ").split()))

    if not validare_input(t, n, margele):
        print("Datele nu corespund restricțiilor impuse.")
    else:
        print(solve(t, n, margele))



</syntaxhighlight>

'''Explicatie cod:'''

Acest cod calculează numărul de mărgele de tipul 1 și 2 și determină numărul maxim de mărgele consecutive de tipuri diferite care pot fi utilizate într-un colier, în funcție de cerințele date.

La început, se citește valoarea lui T (tipul cerinței) și numărul N de mărgele din colier. Apoi, se citesc cele N mărgele. Pentru a valida inputul, se verifică dacă T este 1 sau 2, dacă N este între 1 și 50000 și dacă mărgelele sunt între 12 și 987654321, nu conțin cifra 0 și nu au cifre repetate.

După validarea inputului, se determină mărgelele de tipul 1 și 2 prin verificarea dacă ultima cifră a unei mărgele este mai mare decât prima cifră. Pentru fiecare tip de cerință (T), se calculează răspunsul cerut.

Dacă T este 1, se afișează numărul de mărgele de tipul 1. Dacă T este 2, se determină numărul maxim de mărgele consecutive de tipuri diferite care pot fi utilizate într-un colier. Pentru a face acest lucru, se parcurge lista de mărgele și se determină numărul maxim de mărgele consecutive de tipuri diferite care încep cu fiecare mărgea. Se compară aceste numere pentru a găsi numărul maxim de mărgele consecutive de tipuri diferite care pot fi utilizate în colier.

În cazul în care inputul nu este valid, se afișează un mesaj de eroare.