[백준/Python] 10798번 : 세로읽기

https://www.acmicpc.net/problem/10798

10798번 : 세로읽기

문제 풀이

1. 2차원 배열 형식으로 입력받기

2. 각 줄의 최대 길이를 비교 (해당 문자열의 길이가 최대 길이보다 작다면 넘어갈 수 있도록)

3. 출력할 순서로 정리할 리스트 생성

4-1. 문자열의 길이가 최대길이보다 짧으면서 확인하는 위치가 비어있는 곳이라면 건너뛰기

4-2. 위에서 해당사항이 없다면 출력 리스트에 추가

5. 출력 리스트 출력

문제 코드

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

max_length = 0
for i in range(5):
    if len(arr[i]) > max_length:
        max_length = len(arr[i])
        
toPrint = []
for i in range(max_length):
    for j in range(5):
        if (len(arr[j]) < max_length) and (i >= len(arr[j])):
            continue
        toPrint.append(arr[j][i])
print("".join(toPrint))