[백준/BOJ] 10798번: 세로읽기 (Python) 풀이

Beakjoon Online Judge(BOJ) 의 10798번 문제인 '세로읽기' 를 풀어보았다.

[문제 정보]

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

 

---

문제 개요:

[ 문제 ]

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

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

<그림 1>

 

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

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

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

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

[ 입력 ]

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

[ 출력 ]

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

입력받은 값을 세로로 순차적으로 출력하면 되는 문제인데, 입력받은 단어의 길이가 모두 같으면 상관 없지만

위 예제 같이 단어의 길이가 서로 다른 경우에는, 세로로 출력하는 중에 최대 길이 단어(위 예제 단어 최대 길이: 6)보다 길이가 부족한 단어(afzz, 09121)는 각각 5번째, 6번째 문자를 출력하려 할 때 오류가 발생할 수 있다. (없는 값에 접근하려 했기 때문)

이 점을 유의하며 문제를 해결해야 한다.

필자는 위 문제를, 입력받은 단어 중 제일 긴 단어의 길이만큼, 길이가 부족한 단어의 길이를 특정 문자(예시: !)를 추가하여 늘려주고, 출력하기 전에 특정 문자는 공백으로 치환하는 방법으로 문제를 해결했다.

예시) 제일 긴 단어의 길이는 6이고, 길이가 부족한 'afzz', '09121'을 'afzz!!', '09121!' 처럼 길이를 맞춰 준 후 !는 공백으로 치환한다.

알고리즘:

1. 입력받은 단어를 저장해 둘  wordArr 리스트와, 출력할 단어(세로로 읽은 단어)를 저장할 word 변수, 입력받은 단어의 최대 길이를 저장할 maxLen 변수를 만든다.
   
   
2. 입력은 다섯줄로 고정되어 있으므로, 5번 반복하는 반복문을 만든다. (문자를 입력받고, 최대 길이를 저장하는 역할)
3. 다섯줄의 단어를 리스트 형태로 한 문자씩 끊어서(['A', 'B', 'C', 'D', 'E'] 같은 형태) 순차적으로 입력받고, 입력받은 단어는 바로 wordArr 리스트에 추가하고, 이와 동시에 입력받은 단어 중 최대 길이를 maxLen 변수에 저장한다.
   
   
4. 다시 한번 5번 반복하는 반복문을 만든다. (최대 단어 길이보다 짧은 길이의 단어에 특정 문자(!) 를 추가시켜 길이를 맞추는 역할)
5. 최대 단어 길이보다 짧은 단어의 경우, 최대 단어 길이 - 현재 단어 길이 만큼 반복하는 이중 반복문을 만들어, 짧은 단어에 최대 길이만큼 특정 문자(!)를 추가한다. (예시: afzz(길이 4) → 최대 단어 길이(6) - 현재 단어 길이(4) = 특정 문자 2번 추가)
   
   
6. 최대 단어 길이만큼 반복하는 반복문을 만든다.
7. 5번 반복하는 이중 반복문을 만들어 wordArr에 저장된 단어를 세로로 읽은 형태로 word에 저장한다.
8. replace() 함수를 사용하여 word의 특정 문자를 모두 공백으로 치환한다.
9. 특정 문자가 모두 공백으로 치환되어 정상적으로 세로로 읽은 순서대로 된 word를 출력한다.
   
   

코드:

wordArr = []
word = ''
maxLen = 0

for _ in range(5):
    inputArr = list(input())
    wordArr.append(inputArr)
    
    if len(inputArr) > maxLen:
        maxLen = len(inputArr)

for i in range(5):
    if maxLen > len(wordArr[i]):
        for _ in range(maxLen - len(wordArr[i])):
            wordArr[i].append('!')

for i in range(maxLen):
    for j in range(5):
        word += wordArr[j][i]

word = word.replace('!', '')
print(word)

위 풀이에 대한 질문이나, 틀린 정보가 있다면 댓글로 말씀해 주시면 감사하겠습니다!