[BOJ][⚪2][백준#01283] 단축키 지정

문제 출처

BAEKJOON Online Judge #1283

문제

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

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

입력

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

출력

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

예제

예제 1

입력

5
New
Open
Save
Save As
Save All

출력

[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

출력

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

My Sol

import sys
input = sys.stdin.readline
from collections import deque

def check_in_D(c):
    global D
    if c in D: return True
    if c.upper() in D: return True
    if c.lower() in D: return True
    return False


def check_first(word):
    global D
    if check_in_D(word[0]): return 0, word

    D.add(word[0])
    return 1, f'[{word[0]}]{word[1:]}'


def check_middle(word):
    global D
    wl = len(word)
    for i in range(1, wl):
        if check_in_D(word[i]): continue
        D.add(word[i])
        return 1, f'{word[:i]}[{word[i]}]{word[i+1:]}'
    return 0, word

N = int(input())
D = set()
for _ in range(N):
    done = 0
    words = deque(list(input().split()))
    cur = deque()
    while words:
        word = words.popleft()
        check_ret, word = check_first(word)
        cur.append(word)
        if check_ret:
            done = 1
            break

    if done:
        print(*cur, end=' ')
        print(*words)
        continue

    cur2 = deque()
    while cur:
        word = cur.popleft()
        check_ret, word = check_middle(word)
        cur2.append(word)
        if check_ret: break

    print(*cur2, end=' ')
    print(*cur)

1. 단축키로 등록된 알파벳을 저장하는 D set을 전역에 배치한다.
2. 어떤 알파벳이 D set에 있는지 확인하는 check_in_D 함수를 만든다.
3. 단어의 첫 글자가 D에 있는지 확인하는 check_first 함수를 만든다.
4. 단어의 중간 부분이 D에 들어갈 수 있는지 확인하는 check_middle 함수를 만든다.
5. 각 문장을 입력으로 받는다.
6. 문장을 분할해 check_first와 check_middle 함수를 차례로 통과시킨다. 만약 단축키로 지정할 수 있다면 두 함수에서 자체적으로 단축키로 지정하고 해당 단축키를 D에 저장한 뒤, word에 단축키를 표시해서 반환한다.
7. word들을 모두 모아 출력한다.

결과

맞았습니다!!

모범답안

출처

# empty