0631 - Passwd

Alexandru dorește să devină expert în securitate, iar pentru aceasta s-a apucat să învete mai multe despre siguranța parolelor. El dorește să afle câte parole poate crea folosind a litere mici ale alfabetului englez și b litere mari ale alfabetului englez, c cifre si d caractere din mulțimea {!, @, #, $, %}. Totodată, el vrea să găsească parola cu numărul x în ordine lexicografică, formată din caracterele descrise mai sus.

Cerința

Cunoscând a, b, c, d si x se cere:

a) A x-a parolă în ordine lexicografică, formată din caracterele menționate în enunț.
b) Numărul de parole diferite formate din caracterele menționate în enunț, modulo 666013.

Date de intrare

Fişierul de intrare passwdin.txt conţine pe prima linie un număr natural p. Pentru toate testele de intrare, numărul p poate avea doar valoarea 1 sau valoarea 2.

Pe a doua linie se găsesc patru numere naturale a, b, c și d cu semnificația din enunț, separate prin câte un spațiu. A treia linie conține un singur număr natural x, cu semnificația din enunț.

Date de ieșire

Dacă valoarea lui p este 1, se va rezolva numai punctul a) din cerință.

În acest caz, în fişierul de ieşire passwdout.txt se va scrie parola cu numărul x în ordine lexicografică, descrisă în enunț.

Dacă valoarea lui p este 2, se va rezolva numai punctul b) din cerință.

În acest caz, în fişierul de ieşire passwdout.txt se va scrie numărul de parole distincte, formate cu caracterele descrise în enunț, modulo 666013.

Restricții și precizări

• 1 ≤ a, b, c, d ≤ 2500
• 1 ≤ x ≤ 4000000
• Pentru rezolvarea corectă a primei cerinţe se acordă 20 de puncte, iar pentru cerința a doua se acordă 80 de puncte.
• Ordinea lexicografică a caracterelor este: literele mici, iar apoi cele mari (în ordinea din alfabet), cifrele (în ordine crescătoare), iar apoi caracterele !, @, #, $, %, în această ordine.

Exemplul 1

passwdin.txt

1

1 1 1 1

66

passwdout.txt

aA@5

Explicatie

p = 1

Atenție! Pentru acest test se rezolvă doar cerința a).

Exemplul 2

passwdin.txt

2

1 1 1 1

66

passwdout.txt

145187

Explicatie

p = 1

Atenție! Pentru acest test se rezolvă doar cerința b).

Se observă că valoarea lui x nu este necesară în acest caz.

Rezolvare

#0631 - Passwd
def generate_password(a, b, c, d, x):
    characters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%"
    password = ""
    
    while x > 0:
        for i in range(26):  # literele mici
            for _ in range(a):
                password += characters[i]
                x -= 1
                if x == 0:
                    break
            if x == 0:
                break
        
        for i in range(26, 52):  # literele mari
            for _ in range(b):
                password += characters[i]
                x -= 1
                if x == 0:
                    break
            if x == 0:
                break

        for i in range(52, 62):  # cifrele
            for _ in range(c):
                password += characters[i]
                x -= 1
                if x == 0:
                    break
            if x == 0:
                break

        for i in range(62, 67):  # caracterele speciale
            for _ in range(d):
                password += characters[i]
                x -= 1
                if x == 0:
                    break
            if x == 0:
                break

    return password

def count_distinct_passwords(a, b, c, d):
    modulo_value = 666013
    total_characters = a + b + c + d
    
    if not (1 <= a <= 2500 and 1 <= b <= 2500 and 1 <= c <= 2500 and 1 <= d <= 2500):
        print("Fals")
        return 0

    total_passwords = 1
    
    for i in range(total_characters):
        total_passwords = (total_passwords * (total_characters - i)) % modulo_value

    return total_passwords

# Citirea datelor de intrare
with open("passwdin.txt", "r") as infile:
    p = int(infile.readline().strip())
    a, b, c, d = map(int, infile.readline().split())
    x = int(infile.readline().strip())

# Calculul rezultatului
if p == 1:
    result = generate_password(a, b, c, d, x)
    if result:
        with open("passwdout.txt", "w") as outfile:
            outfile.write(result + '\n')
elif p == 2:
    result = count_distinct_passwords(a, b, c, d)
    if result:
        with open("passwdout.txt", "w") as outfile:
            outfile.write(str(result) + '\n')
else:
    print("Fals")