의석이의 세로로 말해요 Python (SW Expert Academy)

출처 : https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWVWgkP6sQ0DFAUO&categoryId=AWVWgkP6sQ0DFAUO&categoryType=CODE

SW Expert Academy

1. 문제 설명

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

이 장난감에 있는 글자들은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’이다. 의석이는 칠판에 글자들을 수평으로 일렬로 붙여서 단어를 만든다.

다시 그 아래쪽에 글자들을 붙여서 또 다른 단어를 만든다. 이런 식으로 다섯 개의 단어를 만든다. 아래에 의석이가 칠판에 붙여 만든 단어들의 예가 있다.

 

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의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다.

그림 1의 다섯 번째 자리를 세로로 읽으면 D1gk로 읽는다.

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

 

Aa0aPAf985Bz1EhCz2W3D1gkD6x

 

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

 

[입력]
 

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.
 

각 테스트 케이스는 총 다섯 줄로 이루어져 있다.

각 줄에는 길이가 1이상 15이하인 문자열이 주어진다. 각 문자열은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’만으로 이루어져 있다.
 

[출력]
 

각 테스트 케이스마다 #T를 출력하고 한 칸을 띄운 후, 의석이가 세로로 읽은 순서대로 글자들을 출력한다.

 

2. 코드

5개의 단어중 최고 길이를 찾아 그 길이에 맞게 새로운 배열을 만들었습니다. 그 후 다시 입력된 단어를 배열에 넣음으로 모든 단어의 길이를 맞추고 출력을 하였습니다.

굳이 이렇게 하지 않고 짧은 최고 길이만 찾고 바로 출력을 하며 인덱스를 벗어난 경우 실행하지 않도록 하는 코드도 괜찮을 것 같습니다.

 

아래가 작성한 코드입니다.

T = int(input())
for t in range(1, T+1):
    arr = []
    max_len = 0
    # 단어 중 제일 긴 길이를 찾음
    for n in range(5):
        arr.append(input())
        if len(arr[n]) > max_len:
            max_len = len(arr[n])
    # 찾을 길이로 배열 생성
    result_arr = [['' for _ in range(max_len)] for _ in range(max_len)]
    # 생성한 배열에 입력된 단어를 넣음(짧은 길이는 빈 공간 남김)
    for j in range(len(arr)):
        for i in range(len(arr[j])):
            result_arr[j][i] = arr[j][i]
    #세로로 출력
    print('#{} '.format(t), end = '')
    for j in range(len(result_arr)):
        for i in range(len(result_arr[j])):
            print(result_arr[i][j], end = '')
    print()