Editing 1758 - Bile 3 (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
def simulate_balls(m, n, p, obstacles, balls):
    # Inițializăm matricea pentru terenul de joc
    board = [[0] * n for _ in range(m)]
    
    # Adăugăm obstacolele pe tablă
    for obstacle in obstacles:
        x, y = obstacle
        board[x - 1][y - 1] = 1
    
    # Inițializăm vectorul pentru bilele lăsate să cadă de pe prima linie
    first_row_balls = balls
    
    # Iterăm prin fiecare coloană, de la stânga la dreapta
    for j in range(n):
        # Numărăm câte bile au ajuns în fiecare celulă de pe ultima linie
        for i in range(m):
            if i == m - 1:
                # Dacă suntem pe ultima linie, adăugăm numărul de bile lăsate să cadă din coloana j
                board[i][j] += first_row_balls[j]
            else:
                # Dacă nu suntem pe ultima linie, simulăm căderea bilelor
                if board[i][j] > 0:
                    # Bila va cădea în jos până întâlnește un obstacol sau ajunge pe ultima linie
                    k = i + 1
                    while k < m and board[k][j] == 0:
                        k += 1
                    # Dacă am întâlnit un obstacol, bilele se vor deplasa pe orizontală
                    if k < m:
                        if j > 0 and board[k][j - 1] == 1:
                            board[k][j - 1] += board[i][j]
                        if j < n - 1 and board[k][j + 1] == 1:
                            board[k][j + 1] += board[i][j]
                    else:
                        # Dacă am ajuns pe ultima linie, adăugăm bilele în acea celulă
                        board[k - 1][j] += board[i][j]
    
    # Extragem numărul de bile din fiecare celulă de pe ultima linie
    last_row_balls = board[-1]
    
    return last_row_balls

def main():
    # Citim datele de intrare din fișier
    with open("bile.in", "r") as fin:
        m, n, p = map(int, fin.readline().split())
        obstacles = [tuple(map(int, fin.readline().split())) for _ in range(p)]
        balls = [int(fin.readline()) for _ in range(n)]

    # Simulăm căderea bilelor
    last_row_balls = simulate_balls(m, n, p, obstacles, balls)

    # Scriem rezultatul în fișierul de ieșire
    with open("bile.out", "w") as fout:
        fout.write("\n".join(map(str, last_row_balls)) + "\n")

if __name__ == "__main__":
    main()

</syntaxhighlight>