10798번 세로읽기

문제

아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다. 

이 장난감에 있는 글자들은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’이다. 영석이는 칠판에 글자들을 수평으로 일렬로 붙여서 단어를 만든다. 다시 그 아래쪽에 글자들을 붙여서 또 다른 단어를 만든다. 이런 식으로 다섯 개의 단어를 만든다. 아래 그림 1은 영석이가 칠판에 붙여 만든 단어들의 예이다. 

A A B C D D
a f z z 
0 9 1 2 1
a 8 E W g 6
P 5 h 3 k x

<그림 1>

한 줄의 단어는 글자들을 빈칸 없이 연속으로 나열해서 최대 15개의 글자들로 이루어진다. 또한 만들어진 다섯 개의 단어들의 글자 개수는 서로 다를 수 있다. 

심심해진 영석이는 칠판에 만들어진 다섯 개의 단어를 세로로 읽으려 한다. 세로로 읽을 때, 각 단어의 첫 번째 글자들을 위에서 아래로 세로로 읽는다. 다음에 두 번째 글자들을 세로로 읽는다. 이런 식으로 왼쪽에서 오른쪽으로 한 자리씩 이동 하면서 동일한 자리의 글자들을 세로로 읽어 나간다. 위의 그림 1의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다. 그림 1의 다섯 번째 자리를 세로로 읽으면 D1gk로 읽는다. 

그림 1에서 영석이가 세로로 읽은 순서대로 글자들을 공백 없이 출력하면 다음과 같다:

Aa0aPAf985Bz1EhCz2W3D1gkD6x

칠판에 붙여진 단어들이 주어질 때, 영석이가 세로로 읽은 순서대로 글자들을 출력하는 프로그램을 작성하시오.

입력

총 다섯줄의 입력이 주어진다. 각 줄에는 최소 1개, 최대 15개의 글자들이 빈칸 없이 연속으로 주어진다. 주어지는 글자는 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’ 중 하나이다. 각 줄의 시작과 마지막에 빈칸은 없다.

출력

영석이가 세로로 읽은 순서대로 글자들을 출력한다. 이때, 글자들을 공백 없이 연속해서 출력한다. 

예제 입력 1 복사

ABCDE
abcde
01234
FGHIJ
fghij

예제 출력 1 복사

Aa0FfBb1GgCc2HhDd3IiEe4Jj

예제 입력 2 복사

AABCDD
afzz
09121
a8EWg6
P5h3kx

예제 출력 2 복사

Aa0aPAf985Bz1EhCz2W3D1gkD6x

 

matrix = []

for _ in range(5):

    row = list(str(input()))

    row.extend([''] * (15 - len(row)))

    matrix.append(row)

transposed_matrix = [list(row) for row in zip(*matrix)]

for tm in transposed_matrix:

    for t in tm:

        if t != '':

            print(t, end='')

처음에는 다음과 같이 풀었다. max_length가 15니까 15로 하는 것이다. 하지만 이것은 빈문자열이 많이 들어가기 때문에 비효율적이다. 그래서 리스트의 최대 길이를 구해서 전치행렬로 구해주거나, 아니 2중 for문에서 그냥 출력해주기도 할 수 있다. 한번 해보겠다.

 

# 5줄의 입력을 받아 리스트로 변환

matrix = [list(input()) for _ in range(5)]

# 각 줄의 길이를 저장하고 최대 길이를 계산

max_len = max(len(row) for row in matrix)

# 각 줄의 길이를 최대 길이에 맞추어 빈 문자열로 채움

for row in matrix:

    row.extend([''] * (max_len - len(row)))

# 전치된 행렬 생성

transposed_matrix = zip(*matrix)

# 전치된 행렬을 순회하며 빈 문자열이 아닌 문자만 출력

for col in transposed_matrix:

    for char in col:

        if char:

            print(char, end='')

 

다음과 같이 각 길이에 대한 max를 구하면 굳이 15번 전부 loop를 돌지 않아도 된다.!

여기서 배우는 것은 extend를 할때 빈문자열을 하려면 ''을 그냥 써주는 것이 아닌 대괄호로 묶어서 [''] 해야지 list에 추가가 된다. 그리고.! 전치행렬을 만들어주기 위해서는 n*n으로 만들어줘야하고 그리고 나서 zip(*matrix)만 해줘도 된다.!