10809번: 알파벳 찾기 - Python

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

문제

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

입력

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

출력

각각의 알파벳에 대해서, a가 처음 등장하는 위치, b가 처음 등장하는 위치, ... z가 처음 등장하는 위치를 공백으로 구분해서 출력한다.

만약, 어떤 알파벳이 단어에 포함되어 있지 않다면 -1을 출력한다. 단어의 첫 번째 글자는 0번째 위치이고, 두 번째 글자는 1번째 위치이다.

예제 입력 1

baekjoon

예제 출력 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

풀이 1 (index() 활용)

import sys
input = sys.stdin.readline

s = input().rstrip()

for i in range(97, 123):    # a:z의 ASCII 코드값이다. ord('a'), ord('z')로 알아낼 수 있다.
    if chr(i) in s:
        print(s.index(chr(i)), end=' ')
    else:
        print(-1, end=' ')

문제의 핵심 아이디어는 아래 3가지인 것 같다.

1. if-else를 사용한 -1의 처리
2. chr(), ord()를 통한 ascii 코드값의 활용
3. 문자열에서 특정 문자가 처음 나오는 위치를 알려주는 index() 메서드

풀이 2 (find() 활용)

import sys
input = sys.stdin.readline

s = input().rstrip()

for i in range(97, 123):
    print(s.find(chr(i)), end=' ')

풀이 1로 문제를 풀고 다른 풀이들을 찾아보니, find() 메서드를 사용하는 것이 더 적절하다는 것을 알았다.
find() 메서드는 특정 문자가 처음 나온 index를 반환해줄 뿐 아니라, 찾지 못하면 -1을 반환해준다.

배울만한 점 - find(), index()의 차이점

1. find()
   • str에서만 사용 가능 (list, tuple, dictionary에서는 AttributeError 발생)
   • 찾는 문자가 없으면 -1 반환
2. index()
   • str, list, tupe에서 사용 가능 (dictionary에서는 AttributeError 발생)
   • 찾는 문자가 없으면 ValueError 발생