Editing 0572 - Arma (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
def validare(n, m, k, zid, trageri):
    if not 1 <= n <= 100 or not 1 <= m <= 100 or not 1 <= k <= 100:
        return False, "Datele de intrare nu corespund restrictiilor impuse"
    for row in zid:
        for z in row:
            if not 1 <= z < 1000:
                return False, "Datele de intrare nu corespund restrictiilor impuse"
    for t in trageri:
        if not 1 <= t < 1000:
            return False, "Datele de intrare nu corespund restrictiilor impuse"
    return True, "Datele de intrare corespund restrictiilor impuse"


def main():
    print("Introduceți numărul de linii, numărul de coloane și numărul de trageri:")
    n, m, k = map(int, input().split())
    print("Introduceți configurația inițială a zidului:")
    zid = [list(map(int, input().split())) for _ in range(n)]
    print("Introduceți substanțele afectate de fiecare tragere:")
    trageri = list(map(int, input().split()))

    valid, message = validare(n, m, k, zid, trageri)
    print(message)
    if not valid:
        return

    for tragere in trageri:
        for i in range(n):
            for j in range(m):
                if zid[i][j] == tragere:
                    zid[i][j] = 0
        for _ in range(n):
            for i in range(n - 1, 0, -1):
                for j in range(m):
                    if zid[i][j] == 0 and zid[i - 1][j] != 0:
                        zid[i][j], zid[i - 1][j] = zid[i - 1][j], zid[i][j]

    print("Configurația finală a zidului este:")
    for linie in zid:
        print(' '.join(map(str, linie)))


if __name__ == "__main__":
    main()

</syntaxhighlight>