목표
python으로 백준 실버5 티어 이상의 알고리즘 문제 풀기
문제
예전에는 운영체제에서 크로아티아 알파벳을 입력할 수가 없었다. 따라서, 다음과 같이 크로아티아 알파벳을 변경해서 입력했다.
크로아티아 알파벳 변경
č -> c=
ć -> c-
dž -> dz=
đ -> d-
lj -> lj
nj -> nj
š -> s=
ž -> z=
예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.
dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.
입력
첫째 줄에 최대 100글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.
단어는 크로아티아 알파벳으로 이루어져 있다. 문제 설명의 표에 나와있는 알파벳은 변경된 형태로 입력된다.
출력
입력으로 주어진 단어가 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.
#alphabet = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z=']
S = input()
count = 0
i = 0
while i < len(S):
if S[i] == 'c':
if i+1 < len(S):
if S[i+1] == '=':
i += 1
elif S[i+1] == '-':
i += 1
elif S[i] == 'd':
if i+1 < len(S):
if S[i+1] == '-':
i += 1
elif i+2 < len(S):
if S[i+1] == 'z' and S[i+2] == '=':
i += 2
elif S[i] == 'l':
if i+1 < len(S):
if S[i+1] == 'j':
i += 1
elif S[i] == 'n':
if i+1 < len(S):
if S[i+1] == 'j':
i += 1
elif S[i] == 's':
if i+1 < len(S):
if S[i+1] == '=':
i += 1
elif S[i] == 'z':
if i+1 < len(S):
if S[i+1] == '=':
i += 1
count += 1
i += 1
print(count)
우선 알파벳이 한정되어있기 때문에 'c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z='중 c, d, l, n, s, z로 시작하는 문자열을 검사해준다.
예를들어 c로 시작한다면, 'c=' or 'c-' 둘 중 하나이기 때문에 c뒤에 오는 문자와 함께 검사하여 둘 중 하나에 속하는지 검사해주고, 맞다면 두개가 하나의 알파벳이 되기 때문에 루프를 한번 건너뛰어준다. 아니라면 c개인이 하나의 알파벳이 된다.
만약 입력받은 문자열이 c, d, l, n, s, z로 끝난다면, 그 뒤에 오는 문자를 검사하려 하면 런타임에러를 발생시키기 때문에 문자열 전체 길이와 비교하여 index가 초과되지 않는지 미리 검사한다.
word = input()
lst = ['c=','c-','dz=','d-','lj','nj','s=','z=']
for i in lst:
word = word.replace(i, '*')
print(len(word))
좋은 풀이를 검색해 보던 중 위와 같은 훨씬 간결하고 빠른 코드를 발견했다.
파이썬은 문자열이나 리스트를 순회하는 문제를 처리하기에 좋은 언어이지만, 제대로 활용하지 못하여 쓸데없이 길기만 한 코드를 작성한 것 같다. 파이썬의 문자열 처리와 리스트에 대해 조금 더 공부해보면 더 간결하고 빠른 코드를 작성할 수 있을 것 같다.