[백준/python] 10798 세로읽기

문제

 

코드

 

arr = [list(input()) for _ in range(5)]
l=[]
for i in range(5):
    l.append(len(arr[i]))

for i in range(5):
    if max(l)>l[i]:
        for j in range(max(l)-l[i]):
            arr[i].append('&')

t=''
for i in range(max(l)):
    for j in range(5):
        if arr[j][i]!='&':
            t+=arr[j][i]
print(t)

풀이

2차원 리스트를 이용하여 입력받았다. 가장 긴 단어에 맞춰서 덜한단어들은 '&'을 채워줬다. 만약 '&'이 나오면 읽지않고 넘어가고 나머지는 t에 저장한다. 마지막에 t 출력