Vizualizare sursă pentru „3244 - tabla”

== Enunt ==

O tablă de șah de dimensiune '''n x n''' conține pe toate pătrățelele câte o piesă cu una din culorile: alb, negru, roșu, verde sau albastru. Pe tablă nu există 3 piese consecutive pe aceeași linie sau coloană de aceeași culoare. O mutare presupune interschimbarea a două piese alăturate, fie pe o linie, fie pe o coloană. După mutare se acordă punctaj dacă se obțin următoarele situații și nu numai:

*'''3''' piese de aceeași culoare consecutive pe o linie sau o coloană '''= 1''' punct;
*'''3''' piese de aceeași culoare consecutive pe o linie și o coloană '''= 5''' puncte;
La orice situație de mai sus, o piesă în plus de aceeași culoare mai contribuie cu încă 1 la punctaj. De exemplu, 4 piese de aceeași culoare consecutive pe o linie sau o coloană '''= 2''' puncte. Punctajele obținute de piesele interschimbate se adună. Punctajul este 0 dacă nu se obțin măcar 3 piese consecutive de aceeași culoare.

== Cerinta ==

Găsiți cel mai mare punctaj obținut în urma unei singure mutări.

== Date de intrare ==

Fișierul de intrare '''tablain.txt''' conține, pe prima linie, numărul natural '''n''', ce reprezintă dimensiunea tablei. Pe fiecare dintre următoarele '''n''' linii se află câte '''n''' valori separate prin spatii. Valorile posibile pot fi: '''1, 2, 3, 4 și 5'''. Valoarea '''1''' reprezintă piesa de culoare albă, '''2''' piesa de culoare neagră etc.

== Date de iesire ==

Fișierul de ieșire '''tablaout.txt''' va conține cel mai mare punctaj obținut.

== Restrictii si precizari ==

*1 &les; n &les; 20

== Exemplul 1 ==
;tablain.txt
:3
:1 1 2
:2 2 1
:3 4 5

;tablaout.txt
:Datele introduse corespund restrictiilor impuse
:2

== Exemplul 2 ==
;tablain.txt
:3
:0 0 0
:-10 23 3
:1 72 0

:Datele introduse nu corespund restrictiilor impuse


== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def calculeaza_punctaj(tabla, n, i1, j1, i2, j2):
    punctaj = 0

    # Verificare pe linii
    if i1 < n - 2 and tabla[i1][j1] == tabla[i1 + 1][j1] == tabla[i1 + 2][j1]:
        punctaj += 1
    if i2 < n - 2 and tabla[i2][j2] == tabla[i2 + 1][j2] == tabla[i2 + 2][j2]:
        punctaj += 1

    # Verificare pe coloane
    if j1 < n - 2 and tabla[i1][j1] == tabla[i1][j1 + 1] == tabla[i1][j1 + 2]:
        punctaj += 1
    if j2 < n - 2 and tabla[i2][j2] == tabla[i2][j2 + 1] == tabla[i2][j2 + 2]:
        punctaj += 1

    # Verificare pe diagonale
    if i1 < n - 2 and j1 < n - 2 and tabla[i1][j1] == tabla[i1 + 1][j1 + 1] == tabla[i1 + 2][j1 + 2]:
        punctaj += 1
    if i2 < n - 2 and j2 < n - 2 and tabla[i2][j2] == tabla[i2 + 1][j2 + 1] == tabla[i2 + 2][j2 + 2]:
        punctaj += 1

    return punctaj

def cel_mai_mare_punctaj(tabla, n):
    max_punctaj = 0

    for i1 in range(n):
        for j1 in range(n):
            for i2 in range(i1, n):
                for j2 in range(n):
                    # Pentru fiecare pereche de piese alaturate, calculam punctajul
                    if (i1 != i2 or abs(j1 - j2) != 1) and (j1 != j2 or abs(i1 - i2) != 1):
                        tabla[i1][j1], tabla[i2][j2] = tabla[i2][j2], tabla[i1][j1]
                        punctaj = calculeaza_punctaj(tabla, n, i1, j1, i2, j2)
                        max_punctaj = max(max_punctaj, punctaj)
                        tabla[i1][j1], tabla[i2][j2] = tabla[i2][j2], tabla[i1][j1]  # revenire la starea initiala

    return max_punctaj

def verificare_date_intrare(n, tabla):
    if not (1 <= n <= 20):
        return False

    for i in range(n):
        if len(tabla[i]) != n or not all(1 <= tabla[i][j] <= 5 for j in range(n)):
            return False

    return True

# Citire date de intrare
with open("tablain.txt", "r") as file:
    n = int(file.readline())
    tabla = [list(map(int, file.readline().split())) for _ in range(n)]

# Verificare date de intrare
if not verificare_date_intrare(n, tabla):
    print("Datele introduse nu corespund restricțiilor impuse")
else:
    # Calcul si afisare rezultat
    rezultat = cel_mai_mare_punctaj(tabla, n)
    with open("tablaout.txt", "w") as file:
        file.write(str(rezultat) + "\n")


</syntaxhighlight>

== Explicatie ==

Prin interschimbarea elementului (1,3) cu (2,3) se obține:
:1 1 1
:2 2 2
:3 4 5
și am 1 punct din prima linie și 1 punct din a doua linie.