[백준/Python] 1283 단축키 지정

 

📌 문제 탐색하기

• 한글 프로그램의 메뉴에 총 N(1 ≤ N ≤ 30)개의 옵션
• 각 옵션은 한 개 또는 여러 개의 단어로 옵션 기능 설명
• 위에서부터 차례대로 각 옵션에 단축키를 의미하는 대표 알파벳 지정하기
  • 하나의 옵션에 대해 왼쪽에서부터 오른쪽 순서로 단어의 첫 글자가 이미 단축키로 지정되었는지 살펴보기
    • 아직 지정이 안되어있다면 그 알파벳이 단축키
  • 모든 단어의 첫 글자가 이미 지정되어 있다면 왼쪽에서부터 차례대로 알파벳을 보면서 단축키로 지정안된 것을 단축키로 지정
  • 어떠한 것도 단축키로 지정할 수 없다면 그냥 놔두기
  • 대소문자 구분하지 않음
• 각 옵션을 출력하며 단축키로 지정된 알파벳은 좌우에 [] 괄호를 씌워서 표현

---

📌 코드 설계하기

1. 옵션의 개수 N 입력 받기
2. 옵션을 나타내는 문자열 options 리스트로 입력 받기
3. 단축어 알파벳 사용 여부를 체크할 alphabets 리스트 False로 초기화
4. 각 옵션의 단축어 위치를 기록할 shortcut_idx 리스트 0으로 초기화
5. 옵션 하나마다 단축어 지정 규칙에 따라 단축어를 지정한다.
   1. 옵션의 단어들 첫글자를 탐색하며, alphabets에서 아직 사용하지 않았다고 체크된 글자이면 사용 체크 후에, 단축어 위치를 기록한다.
   2. 모든 단어의 첫글자를 탐색했는데 여전히 shortcut_idx[cur]이 -1이라면, 앞에서부터 차례대로 모든 글자를 확인하면서, alphabets에서 아직 사용하지 않은 글자가 나올때까지 탐색한다.
6. 모든 옵션에 대한 단축어 지정이 끝나면, shortcut_idx에 따라 각 옵션 문자열의 단축어에 [ ]를 추가하여 출력한다.

---

📌 정답 코드

N = int(input())
options = [input().split() for _ in range(N)]

alphabets = [False]*26
shortcuts_idx = [0]*N

def check_shortcut(option_idx, word_idx, cur_alphabet):
  cur_alphabet = ord(cur_alphabet.lower())
  cur_alphabet = cur_alphabet - 97
  if not alphabets[cur_alphabet]:
    shortcuts_idx[option_idx] = word_idx
    alphabets[cur_alphabet] = True
    return True
  return False

for n in range(N):
  option = options[n]
  isFound = False

  # 각 단어 첫글자 탐색
  for i in range(len(option)):
    if check_shortcut(n, (i, 0), option[i][0]):
      isFound = True
      break
    
  # 단어 첫글자에서 단축어를 지정 못했다면 모든 글자
  if not isFound:
    for i in range(len(option)):
      for j in range(len(option[i])):
        if check_shortcut(n, (i, j), option[i][j]):
          isFound = True
          break
      if isFound:
        break

for n in range(N):
  option = options[n]
  if not shortcuts_idx[n]:
    print(*option)
  else:
    words = ""
    for i in range(len(option)):
      for j in range(len(option[i])):
        if shortcuts_idx[n] == (i, j):
          words += f'[{option[i][j]}]'
        else:
          words += option[i][j]
      words += " "
    print(words)