0739 - Cuburi 1

Cerința

Fie n cuburi de aceeaşi mărime, cu feţe colorate. Culorile sunt codificate prin câte o literă de la A la M. Pentru fiecare cub se cunosc culorile feţelor în ordinea: bază, capac, faţă frontală, faţă laterală dreapta, faţa din spate, faţă laterală stânga. Să se determine numărul maxim de cuburi care, răsturnate şi rotite convenabil, pot fi puse unul peste altul astfel încât să formeze un turn cu toate feţele uniform colorate (fiecare faţă a turnului sa fie de aceeaşi culoare, de la primul, până la ultimul cub al turnului).

Date de intrare

Fișierul de intrare cuburi1IN.txt conține pe prima linie numărul de cuburi n, iar pe a următoarele n linii culorile fețelor celor n cuburi.

Date de ieșire

Fișierul de ieșire cuburi1OUt.txt va conține pe prima linie un singur număr ce reprezintă numărul maxim de cuburi care, răsturnate şi rotite convenabil, pot fi puse unul peste altul astfel încât să formeze un turn cu feţele uniform colorate. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

Restricții și precizări

• 0 ≤ n ≤ 50000
• cuburile ce formează un turn sunt aşezate numai unul peste celalalt, nu şi unul lângă celălalt;
• culorile fetelor unui cub se pot repeta pentru două sau mai multe dintre cele 6 feţe ale sale;
• orice cub poate fi rotit sau răsturnat pentru a fi adus într-o poziţie convenabilă;
• culorile fetelor cuburilor care nu formează feţele laterale ale turnului nu au nici o importanţă;
• culorile fețelor sunt litere mari din mulțimea {A,B,...,M}

Exemplu 1:

cuburi1IN.txt

3
ACADEB
FBCDAE
AEDCBB

cuburi1OUt.txt

2

Exemplu 2:

cuburi1IN.txt

3000000000
ACADEB
FBCDAE
AEDCBB

cuburi1OUt.txt

Datele nu corespund restrictiilor impuse

Explicație

Primul cub poate fi păstrat în poziţia sa, având feţele laterale A, D, E, B (frontal, lateral-dreapta, spate, lateral-stânga), iar al treilea cub poate fi răsturnat astfel încât să aibă capacele B şi C, şi rotit astfel încât să aibă feţele laterale să fie tot A, D, E, B.Rezolvare

Rezolvare

from collections import defaultdict

def f(a, b, c, d, i, v, verif):
    if c < a:
        a, c = c, a
    if d < b:
        b, d = d, b
    if b < a or (b == a and d < c):
        a, b = b, a
        c, d = d, c
    rez = a + b + c + d
    if verif[rez] != i:
        verif[rez] = i
        v[rez] += 1
        return v[rez]
    verif[rez] = i
    return 0

def main():
    try:
        with open("cuburi1IN.txt", "r") as infile:
            n = int(infile.readline().strip())
            if not (0 <= n <= 50000):
                raise ValueError
            
            v = defaultdict(int)
            verif = defaultdict(int)
            mx = 0

            for i in range(1, n + 1):
                c = infile.readline().strip()
                if len(c) != 6 or not all('A' <= char <= 'M' for char in c):
                    raise ValueError
                
                mx = max(mx, f(c[2], c[3], c[4], c[5], i, v, verif))
                mx = max(mx, f(c[0], c[3], c[1], c[5], i, v, verif))
                mx = max(mx, f(c[0], c[2], c[1], c[4], i, v, verif))
        
        with open("cuburi1OUT.txt", "w") as outfile:
            outfile.write(str(mx))
    
    except ValueError:
        with open("cuburi1OUT.txt", "w") as outfile:
            outfile.write("Datele nu corespund restrictiilor impuse")

if __name__ == "__main__":
    main()