【題解】ZeroJudge e971: 2. 梗圖著色 (Coloring)

【題目敘述】https://zerojudge.tw/ShowProblem?problemid=e971

Python code (credit: Amy Chou)

m, n = map(int, input().split())
grid = []
for i in range(m):
    grid.append(list(map(str, input().split())))
    
for i in range(m):
    l = -1
    for j in range(n):
        if grid[i][j] == "1":
            if l < 0:
                l = j
            else:
                for k in range(l+1, j):
                    grid[i][k] = "1"
                l = -1

for i in range(m):
    print(" ".join(grid[i]))       
分享本文 Share with friends