백준 알고리즘 10709번 기상캐스터(python)

단순 구현 문제이다. 처음엔 아래 코드처럼 풀고

H, W = map(int, input().split())
res = []
for _ in range(H):
    s = input()    
    li = []
    c_idx = -1
    for i in range(W):
        if s[i] == 'c':
            li.append(0)
            c_idx = i
        elif c_idx == -1:
            li.append(-1)
        else:
            li.append(i-c_idx)
    res.append(li)
for a in res:
    print(*a)

마지막에 이렇게 고쳤다. 둘 다 통과됐다.

H, W = map(int, input().split())
for _ in range(H):
    s = input()    
    c_idx = -1
    for i in range(W):
        if s[i] == 'c':
            c_idx = i
            print(0, end=' ')
        elif c_idx == -1:
            print(-1, end=' ')
        else:
            print(i-c_idx, end=' ')
    print()