백준 1283번. 단축키 지정 (Python / 파이썬)

문제

한글 프로그램의 메뉴에는 총 N개의 옵션이 있다. 각 옵션들은 한 개 또는 여러 개의 단어로 옵션의 기능을 설명하여 놓았다. 그리고 우리는 위에서부터 차례대로 각 옵션에 단축키를 의미하는 대표 알파벳을 지정하기로 하였다. 단축키를 지정하는 법은 아래의 순서를 따른다.
1. 먼저 하나의 옵션에 대해 왼쪽에서부터 오른쪽 순서로 단어의 첫 글자가 이미 단축키로 지정되었는지 살펴본다. 만약 단축키로 아직 지정이 안 되어있다면 그 알파벳을 단축키로 지정한다.
2. 만약 모든 단어의 첫 글자가 이미 지정이 되어있다면 왼쪽에서부터 차례대로 알파벳을 보면서 단축키로 지정 안 된 것이 있다면 단축키로 지정한다.
3. 어떠한 것도 단축키로 지정할 수 없다면 그냥 놔두며 대소문자를 구분치 않는다.
4. 위의 규칙을 첫 번째 옵션부터 N번째 옵션까지 차례대로 적용한다.

입력

첫째 줄에 옵션의 개수 N(1 ≤ N ≤ 30)이 주어진다. 둘째 줄부터 N+1번째 줄까지 각 줄에 옵션을 나타내는 문자열이 입력되는데 하나의 옵션은 5개 이하의 단어로 표현되며, 각 단어 역시 10개 이하의 알파벳으로 표현된다. 단어는 공백 한 칸으로 구분되어져 있다.

출력

N개의 줄에 각 옵션을 출력하는데 단축키로 지정된 알파벳은 좌우에 [ ] 괄호를 씌워서 표현한다.

---

문제 풀이 

이 문제는 문제에서 제시한 요구사항을 그대로 구현하기만 하면 되는 문제이다. 

하지만 세부적으로 고려해야 할 사항이 많아서 하나라도 놓치면 틀리기 쉬운 문제였다. 

 

Step 1. 문장을 단어로 쪼갠다. 

단축키를 지정할 때 가장 먼저 각 단어의 첫글자 부터 확인해야 한다. 따라서 문장을 단어로 쪼개어 리스트로 저장해주었다. 

 

Step 2. 각 단어의 첫글자가 단축키인지 확인한다. 

첫번째 단어부터 각 단어의 첫글자들을 탐색하며 단축키로 지정되어있는지 확인하고, 만약 단축키로 지정되었지 않는다면 단축키로 지정해준뒤 출력해준다. 그리고 단축키로 지정되었다고 표시하기 위해 flag 변수를 사용하였다. 이때 주의해야할 점은 단축키를 지정할 때 대소문자 구분을 하지 않는다고 했으므로 모두 대문자로 단축키 리스트에 저장해준다. 

 

Step 3. flag가 False라면 각 단어의 모든 글자를 탐색하며 단축키인지 확인한다. 

첫번째 단어부터 각 단어의 모든 글자들을 탐색하며 단축키로 지정되어있지 않다면 단축키로 지정해준뒤 출력해준다. 마찬가지로 flag 변수를 사용하였고 for문을 중단시키기 위해 check 변수도 사용하였다. 

 

Step 4. flag가 False라면 모든 글자가 단축키가 될 수 없다는 것이므로 그대로 출력한다. 

 

이 모든 과정에서 쪼갠 단어 리스트를 다시 하나의 문장으로 출력하기 위해서 join 함수를 이용하였다. join 함수는 리스트에 문자열이 포함되어 있을 때 하나의 문자열로 합쳐서 출력할 수 있게 해준다. 

---

My Code

n = int(input())

exist = []
for _ in range(n):
    words = list(input().split())

    flag = False 
    for i in range(len(words)):
        if words[i][0].upper() not in exist:
            exist.append(words[i][0].upper())
            flag = True 
            words[i] = '[' + words[i][0] + ']' + words[i][1:]
            print(' '.join(words)) 
            break 
    
    if not flag:
        for i in range(len(words)):
            check = False 
            for j in range(len(words[i])):
                if words[i][j].upper() not in exist:
                    exist.append(words[i][j].upper())
                    flag = True 
                    check = True 
                    words[i] = words[i][:j] + '[' + words[i][j] + ']' + words[i][j+1:]
                    print(' '.join(words))
                    break 
            if check: break 
    
    if not flag:
        print(' '.join(words))