[백준] #1283 단축키 지정(python)

📌문제

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

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

입력

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

출력

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

예제 입력1

5
New
Open
Save
Save As
Save All

예제 출력1

[N]ew
[O]pen
[S]ave
Save [A]s
Sa[v]e All

예제 입력2

8
New window
New file
Copy
Undo
Format
Font
Cut
Paste

예제 출력2

[N]ew window
New [f]ile
[C]opy
[U]ndo
F[o]rmat
Fon[t]
Cut
[P]aste

백준 1283번 문제

💡Idea

단축키를 어떻게 저장할지 고민을 했던 문제입니다🤔

저는 집합(set)을 이용하여 단축키를 저장하는 방식으로 문제를 해결하였습니다.
in 연산자를 사용할 때 시간복잡도가 $O(1)$이기 때문에, 집합을 사용하면 이전에 저장된 단축키를 확인할 때 시간을 줄일 수 있습니다.

💻코드

• ⏰ 시간 : 68 ms / 메모리 : 30840 KB

import sys
input = sys.stdin.readline
N = int(input())
keys = set() # 단축키 저장 집합

def shortcut(option): # option을 입력하면 단축키가 적용된 option return
    for i in range(len(option)): # 순서 1 : 단어의 첫 글자 확인
        if option[i][0].lower() not in keys or option[i][0].upper() not in keys:
        	# 해당 알파벳의 소문자, 대문자가 keys에 없다면 추가 후 [] 씌워 return
            keys.add(option[i][0].lower())
            keys.add(option[i][0].upper())
            option[i] = '[' + option[i][0] + ']' + option[i][1:]
            return option
    for i in range(len(option)):  # 순서 2, 3 : 첫 글자 제외 모든 알파벳 확인 
        for j in range(1, len(option[i])):
            if option[i][j].lower() not in keys or option[i][j].upper() not in keys:
            	# 해당 알파벳의 소문자, 대문자가 keys에 없다면 추가 후 [] 씌워 return
                keys.add(option[i][j].lower())
                keys.add(option[i][j].upper())
                option[i] = option[i][:j] + '[' + option[i][j] + ']' + option[i][j+1:]
                return option
    return option

for _ in range(N):
    print(' '.join(shortcut(input().split())))