1. 문제 설명
[문제 내용]
아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다.
이 장난감에 있는 글자들은 영어 대문자 ‘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’ 중 하나이다. 각 줄의 시작과 마지막에 빈칸은 없다.
[출력]
영석이가 세로로 읽은 순서대로 글자들을 출력한다. 이때, 글자들을 공백 없이 연속해서 출력한다.
[입출력 예]
2. 풀이
words = [input() for _ in range(5)]
max_len = max(len(word) for word in words)
result = []
for col in range(max_len):
for row in range(5):
if col < len(words[row]): # 현재 열(col)에 글자가 존재하면
result.append(words[row][col]) # 해당 글자를 결과 리스트에 추가한다
print(''.join(result))3. 회고
3-1. 문제 해결 과정
행렬로 풀면 되겠다는 생각은 했는데 저장하고 어떻게 꺼내올 것인지가 문제였다. 2차원 배열로 접근하다가 너무 복잡해져서 이게 맞나 고민하다가 더 미궁 속으로 빠져든 것 같다. 알고보니 for 문을 사용해서 리스트에 있는 것을 변수에 할당하면 되는 거였다.
col과 row가 각각 max_len, len(words[row])인 이유는 입력 받은 것을 세로로 읽어 나가기 때문에 행과 열이 바뀌어서 그렇다. 현재 열에 글자가 존재하면 해당 글자를 결과 리스트에 추가하면서, 최종적으로는 join()을 이용하여 공백 없이 한번에 출력해 나가면 된다.
3-2. 새롭게 배운 내용
max(len(word) for word in words): 현재words안에 우리가for문으로 받아온 5개의 내용이 리스트 형태로 들어있다. 이를word변수를 이용하여 순회하면서 리스트 안의 각 내용들의 길이를 계산한다. 마지막으로max()가 있으니 그 중에서 최대 길이를 가진 값이max_len에 저장된다. 이전에for _ in range()와 같은 내용이다.
