[Python/파이썬] 백준 1283번 단축키 지정

1283번: 단축키 지정

 

문제

한글 프로그램의 메뉴에는 총 N개의 옵션이 있다. 각 옵션들은 한 개 또는 여러 개의 단어로 옵션의 기능을 설명하여 놓았다. 그리고 우리는 위에서부터 차례대로 각 옵션에 단축키를 의미하는 대표 알파벳을 지정하기로 하였다. 단축키를 지정하는 법은 아래의 순서를 따른다.

1. 먼저 하나의 옵션에 대해 왼쪽에서부터 오른쪽 순서로 단어의 첫 글자가 이미 단축키로 지정되었는지 살펴본다. 만약 단축키로 아직 지정이 안 되어있다면 그 알파벳을 단축키로 지정한다.
2. 만약 모든 단어의 첫 글자가 이미 지정이 되어있다면 왼쪽에서부터 차례대로 알파벳을 보면서 단축키로 지정 안 된 것이 있다면 단축키로 지정한다.
3. 어떠한 것도 단축키로 지정할 수 없다면 그냥 놔두며 대소문자를 구분치 않는다.
4. 위의 규칙을 첫 번째 옵션부터 N번째 옵션까지 차례대로 적용한다.

입력

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

출력

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

코드

n = int(input())

short_keys = set()

for _ in range(n):
    option = input()
    word_lst = list(option.strip().split())
    flag = False
    for i in range(len(word_lst)):
        if word_lst[i][0].lower() not in short_keys:
            short_keys.add(word_lst[i][0].lower())
            for j in range(len(word_lst)):
                if j != i:
                    print(word_lst[j], end=" ")
                else:
                    print("["+word_lst[j][0]+"]"+word_lst[j][1:], end=" ")
            print()
            flag = True
            break
    if not flag:
        for i in range(len(option)):
            if option[i].lower() not in short_keys and option[i] != ' ':
                short_keys.add(option[i].lower())
                print(option[:i]+"["+option[i]+"]"+option[i+1:])
                flag = True
                break
    if not flag:
        print(option)

우선 옵션의 각 단어의 첫 글자를 단축키로 지정할 수 있는지 알아보고, 안된다면 왼쪽에서부터 한 글자씩 살펴보며 지정이 되는지 확인한다. 어떠한 글자도 단축키로 지정할 수 없다면 처음에 입력된 옵션을 그대로 출력해주면 된다.