세로읽기

import sys


def main():
    input = sys.stdin.readline

    lines = [input().rstrip() for _ in range(5)]

    cols = max(len(line) for line in lines)
    result = []
    for i in range(cols):
        for line in lines:
            if i < len(line):
                result.append(line[i])

    print("".join(result))


main()

for 문의 순서를 조정하여 열 방향을 우선 순회하도록 하였습니다. 한 줄의 길이를 넘은 열은 무시하도록 if문을 사용하였습니다.

댓글