View source for 0599 - Covoare


= Cerința =
Se consideră o încăpere de lungime <code>m</code> și lățime <code>n</code> împărțită în <code>n*m</code> zone pătrate, sub forma unei matrice cu <code>n</code> linii și <code>m</code> coloane. Încăperea este acoperită în totalitate cu <code>p</code> covoare de diferite dimensiuni și culori astfel încât acestea nu se suprapun și încăperea este acoperită în totalitate. Mai mult fiecare zonă pătrată a încăperii este acoperită cu un singur covor.

Gigel, administratorul clădirii, este responsabil cu spălarea covoarelor. Astfel el a notat, în ordine, dimensiunile și culoarea fiecărui covor, în ordine, de sus în jos și de la stânga la dreapta. Covoarele au fost scoase și spălate, dar acum Gigel vă cere ajutorul.

Cunoscând dimensiunile încăperii, numărul de covoare precum și dimensiunile și culoarea fiecărui covor, să se refacă așezarea inițială a acestora.

= Date de intrare =
Fișierul de intrare <code>covoare.in</code> conține pe prima linie numerele <code>n m p</code>, iar pe a următoarele <code>p</code> linii câte trei numere <code>a b c</code> reprezentând lungimea, lățimea și culoarea fiecărui covor.

= Date de ieșire =
Fișierul de ieșire <code>covoare.out</code> va conține o matrice <code>A</code> cu <code>n</code> linii și <code>m</code> coloane, <code>A[i][j]</code> reprezentând culoarea covorului care acoperă zona <code>i j</code>. Fiecare linie a matricei va fi scrisă pe o linie a fișierului, elementele unei linii fiind separate prin exact un spațiu.

= Restricții și precizări =

* <code>1 ≤ n, m ≤ 100</code>
* covoarele trebuie așezate în ordinea dată, și nu pot fi rotite
* culoarea unui covor este un număr natural nenul

= Exemplu: =
<code>covoare.in</code>
 5 6 8
 3 2 2
 1 1 1
 2 3 2
 1 1 4
 4 2 3
 2 2 1
 2 1 1
 2 1 3
<code>covoare.out</code>
 2 2 2 1 2 2
 2 2 2 4 2 2
 3 3 3 3 2 2
 3 3 3 3 1 1
 1 1 3 3 1 1

== Rezolvare ==
<syntaxhighlight lang="python3">
def read_input(n, m, p):
    carpets = []
    for _ in range(p):
        a, b, c = map(int, input().split())
        carpets.append((a, b, c))
    return carpets

def is_valid_placement(carpets, n, m):
    used_area = [[False for _ in range(m)] for _ in range(n)]
    for a, b, _ in carpets:
        for i in range(n - a + 1):
            for j in range(m - b + 1):
                if used_area[i][j]:
                    return False
                for x in range(a):
                    for y in range(b):
                        used_area[i + x][j + y] = True

    return True

def place_carpets(carpets, n, m):
    """Arranges the carpets on the floor."""
    floor = [[0 for _ in range(m)] for _ in range(n)]
    for a, b, c in carpets:
        for i in range(n - a + 1):
            for j in range(m - b + 1):
                if not used_area[i][j]:
                    for x in range(a):
                        for y in range(b):
                            floor[i + x][j + y] = c
                            used_area[i + x][j + y] = True
                    break
    return floor

def print_floor(floor, n, m):
    for i in range(n):
        for j in range(m):
            print(floor[i][j], end=" ")
        print()

def main():
    n, m, p = map(int, input().split())
    carpets = read_input(n, m, p)

    if not is_valid_placement(carpets, n, m):
        print("IMPOSIBIL")
        return

    floor = place_carpets(carpets, n, m)
    print_floor(floor, n, m)

if __name__ == "__main__":
    main()

</syntaxhighlight>