Editing 0599 - Covoare (section)

= 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>