[백준/파이썬]10798번 세로읽기

 

-코드

arr = []
r = []
MAX = 0
for _ in range(5):
    s = list(input())
    if MAX < len(s):
        MAX = len(s)
    arr.append(s)

for i in range(MAX):
    for a in arr:
        if len(a) > i:
            r.append(a[i])
print("".join(r))

처음에는 join을 사용하기 위해 따로 배열을 만들어서 저장하고 출력하려 했으나 2번째 입 출력에서 오류가 떠서 if문으로 continue를 사용해 봤지만 해결이 되지 않아서 찾아보니 그냥 if의 범위를 잘못 설정을 한 것이었습니다.