Editing 2105 - More Time (section)

==Rezolvare==
<syntaxhighlight lang="python" line="1">
#2105 More time
def conditii(clienti, clienti_premium):
    return all((
        (0 <= client['fond'] <= 100_000 for client in clienti),
        (2 <= len(str(client['idx'])) <= 5 for client in clienti),
        (1 <= len(clienti_premium) <= len(clienti) <= 10_000)
    ))


def is_premium(client_idx):
    sir_client_idx = str(client_idx)
    # Un client este premium dacă primul și ultimul caracter din ID-ul său sunt egale
    return sir_client_idx[0] == sir_client_idx[-1]


def moretime(clienti_premium):
    clienti_premium_nr = len(clienti_premium)

    # Inițializăm o listă pentru a ține cont de suma acumulată a fondurilor pentru fiecare client premium
    suma = [0] * (clienti_premium_nr + 1)

    # Iterăm prin fiecare client premium și actualizăm suma acumulată
    for i in range(clienti_premium_nr):
        # Suma acumulată pentru clientul i+1 este suma acumulată pentru clientul i plus fondul clientului i
        suma[i + 1] = suma[i] + clienti_premium[i]['fond']

        # Dacă suma acumulată este divizibilă cu numărul de clienți premium...
        if suma[i + 1] % clienti_premium_nr == 0:
            # Printăm numărul de clienți din ciclu și indicii lor, apoi ieșim din program
            print(i + 1)
            for j in range(i + 1):
                print(clienti_premium[j]['idx'], end=' ')
            exit()

    # Dacă nu există cicluri cu toți clienții premium, găsim cel mai lung ciclu cu doi clienți premium
    i_nou, j_nou = -1, -1
    for i in range(clienti_premium_nr):
        for j in range(i + 1, clienti_premium_nr):
            # Verificăm dacă diferența dintre suma acumulată celor doi clienți este divizibilă cu numărul de clienți premium
            if suma[i + 1] % clienti_premium_nr == suma[j + 1] % clienti_premium_nr:
                i_nou = i
                j_nou = j

    # Printăm lungimea celui mai lung ciclu cu doi clienți premium și indicii lor
    print(j_nou - i_nou)
    for i in range(i_nou + 1, j_nou + 1):
        print(clienti_premium[i]['idx'], end=' ')


if __name__ == "__main__":
    n = int(input())
    clienti = []
    for i in range(n):
        idx, fond = input().split()
        clienti.append({'idx': int(idx), 'fond': int(fond)})

    clienti_premium = []
    for i in range(n):
        if is_premium(clienti[i]['idx']):
            clienti_premium.append({'idx': clienti[i]['idx'], 'fond': clienti[i]['fond']})

    if not conditii(clienti, clienti_premium):
        print("Datele de intrare nu corespund restricțiilor impuse.")
    else:
        print("Datele de intrare corespund restricțiilor impuse.")
        moretime(clienti_premium)

</syntaxhighlight>