BAEKJOON 1316

11 Jan 2020

문제

그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, kin은 k, i, n이 연속해서 나타나기 때문에 그룹 단어이지만, aabbbccb는 b가 떨어져서 나타나기 때문에 그룹 단어가 아니다.

입력

첫째 줄에 단어의 개수 N이 들어온다. N은 100보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 단어가 들어온다. 단어는 알파벳 소문자로만 되어있고 중복되지 않으며, 길이는 최대 100이다.

나의 처음 코드

input_str = []
for i in range(0, int(input())):
    input_str.append(input())

min_index = 0  # 문자열에서 특정 문자가 위치해있는 left most index
max_index = 0  # 문자열에서 특정 문자가 위치해있는 rigth most index
numOfgroup = 0  # 그룹단어 갯수

for input_str in input_str:
    numOfAlpha = 0  # 문자열에서 총 특정 알파벳의 총 갯수
    alphabet = input_str[0]  # 찾는 특정 문자
    while True:
        min_index = input_str.find(alphabet)
        max_index = input_str.rfind(alphabet)
        numOfAlpha = input_str.count(alphabet)
        if (max_index - min_index + 1) != numOfAlpha:
            # 그룹체커가 아님을 확인
            break
        else:
            if max_index + 1 == len(input_str):
                # 그룹체커임을 확인
                numOfgroup += 1
                break
            else:
                alphabet = input_str[max_index + 1]
print(numOfgroup)

나의 아이디어

제일 처음에는 연속 중복되는 알파벳을 1개로 만든 후에, 줄인 문자의 갯수를 카운트 했을 경우 1이 아니면, 그룹체커가 아님을 이용하려 했다. (할지몰라서 못하므) e.g) aaabba 이면 aba로 줄여 a가 2개이므로, 이것은 그룹단어가 아님을 보이려고 했다. 따라서 다른 방법인 문자열이 연속함을 이용하여, 찾고자하는 특정 문자의 갯수와 특정 문자의 left most index와 right most index의 차이가 다르면 그룹체커가 아님을 이용

다른 아이디어

특정문자 Index 찾기(find, rfind)

def rfind(self, sub: Text, start: Optional[int]=..., end: Optional[int]=...): Return the highest index in S where substring sub is found, such that sub is contained within S[start:end]
def find(self, sub: Text, start: Optional[int]=..., end: Optional[int]=...): Return the highest index in S where substring sub is found, such that sub is contained within S[start:end]

정렬하기(sorted)

def sorted(iterable: Iterable[_T], key: Optional[Callable[[_T], Any]]=..., reverse: bool=...): default는 오름차순으로 정렬하여, list로 반환한다. 또한, custom key를 사용하여 sort order를 정할 수 있다.

sorted 함수(+ key)를 이용해서 original 문자열과 sort된 문자열의 비교를 통해, 더 간결한 코딩을 할 수 있다??!
당신도 한번 생각해 보길 바란다. :)

정답

input_str = []
numOfgroup = 0

for i in range(0, int(input())):
    input_str.append(input())

for input_str in input_str:
    if list(input_str) == sorted(input_str, key=input_str.find):
        numOfgroup += 1

print(numOfgroup)

sorted함수 key 예제

students = [
    ('kim', 'A', 100),
    ('park', 'A+', 120),
    ('lee', 'B', 95)
]
sorted(students, key = lambda input: output[2])

key를 통해서 쉽게 원하는 원소의 원하는 방향으로 쉽게 코딩할 수 있다는 점을 알아두자!