백준 10809번: find 메서드는 인덱스만 반환하는가?

문제

 알파벳 소문자로만 이루어진 단어 S가 주어진다. 각각의 알파벳에 대해서, 단어에 포함되어 있는 경우에는 처음 등장하는 위치를, 포함되어 있지 않은 경우에는 -1을 출력하는 프로그램을 작성하시오.

 

입력

 첫째 줄에 단어 S가 주어진다. 단어의 길이는 100을 넘지 않으며, 알파벳 소문자로만 이루어져 있다.

baekjoon

 

출력

 각각의 알파벳에 대해서, a가 처음 등장하는 위치, b가 처음 등장하는 위치, ... z가 처음 등장하는 위치를 공백으로 구분해서 출력한다. 만약, 어떤 알파벳이 단어에 포함되어 있지 않다면 -1을 출력한다. 단어의 첫 번째 글자는 0번째 위치이고, 두 번째 글자는 1번째 위치이다.

1 0 -1 -1 2 -1 -1 -1 -1 4 3 -1 -1 7 5 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1

 

첫 시도

 첫째 줄에 단어 S를 입력받고, result라는 빈 문자열을 생성한다. 알파벳 소문자의 유니코드를 이용해서 for문을 만들고 find 메서드를 사용해 S에서 인덱스 값을 리턴해 result에 추가한다. 마지막으로 join 메서드를 통해 문자 사이 공백을 넣어준다.

import sys

S = sys.stdin.readline().strip()
result = ""

for i in range(ord('a'), ord('z') + 1):
    temp = S.find(chr(i))
    result += str(temp)

print(' '.join(result).strip())

1 0 - 1 - 1 2 - 1 - 1 - 1 - 1 4 3 - 1 - 1 7 5 - 1 - 1 - 1 - 1 - 1 - 1 - 1 - 1 - 1 - 1 - 1

 

최종 코드

 인덱스를 반환하니까 문자열에 저장해도 된다고 단순하게 생각했다. 그런데 find 메서드는 찾고자 하는 문자열이 없는 경우 인덱스가 아닌 음수(-1)를 반환하기 때문에 문자열이 아닌 리스트에 저장해야 했다. 비로소 join 메서드가 원하는 방향으로 잘 작동하게 되었다.

import sys

S = sys.stdin.readline().strip()
result = []

for i in range(ord('a'), ord('z') + 1):
    temp = S.find(chr(i))
    result.append(str(temp))

print(' '.join(result))

1 0 -1 -1 2 -1 -1 -1 -1 4 3 -1 -1 7 5 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1