[Python | 백준 1380번] 귀걸이

https://www.acmicpc.net/problem/1380

1380번: 귀걸이

 

문제

파스칼 고등학교에 다니는 많은 여학생들은 규정에 없는 귀걸이를 착용한 채 도망다닙니다. Sneddon 교감선생님은 흔들거리는 긴 빨간 귀걸이들을 볼때마다 압수합니다.

교감선생님은 귀걸이를 압수당한 여학생들을 숫자를 매겨 리스트를 작성하고 있습니다. 그리고 압수한 귀걸이 뒤쪽에 여학생 번호와 마음대로 선택한 'A' 또는 'B'를 함께 적어두었습니다.

모든 정규 일과와 방과후 수업의 감금이 끝나면, 여학생들은 교감선생님을 찾아와 귀걸이를 돌려받습니다. 불행하게도 어느 날, 교감선생님은 귀걸이가 든 봉투를 잃어버렸고, 하나를 끝내 찾지 못했습니다.

귀걸이를 받지 못해 화난 소녀의 이름을 교감선생님께 알려주세요.

입력

입력은 번호를 가진 시나리오들로 구성됩니다. 시나리오 번호는 1부터 순서대로 증가하고, 각 시나리오는 아래의 내용을 포함합니다.

• 한 줄에 귀걸이를 압수당한 여학생의 수, n (1 ≤ n ≤ 100)이 주어집니다.
• 다음 n줄에 걸쳐 여학생들의 이름(최대 60자)이 주어집니다.
• 다음 2n − 1줄에 여학생 번호와 'A' 또는 'B'가 공백을 사이에 두고 주어집니다. 번호는 교감선생님의 여학생 이름 리스트와 순서가 일치합니다. 즉, 1은 첫 번째로 압수당한 여학생입니다. 여학생 번호는 최대 2번 등장하며, 두 번째로 등장할 때는 첫 번째 경우와 다른 'A' 또는 'B'가 뒤에 적힙니다.
  번호가 처음 등장하는 것은 압수되었음을, 두 번째로 등장하는 것은 돌려받았음을 의미합니다.

'0'을 마지막 줄로 하여 입력이 종료됩니다. '0'은 처리하지 않습니다.

출력

시나리오 번호와 귀걸이를 돌려받지 못한 여학생의 이름을 공백으로 구분하여 한 줄씩 출력하십시오.

 

---

내 풀이

# 31120kb 52ms

 

- 시나리오는 0이 입력될까지 반복이니 while문 사용

- 각 for문을 돌며 학생 이름을 넣고, 1 A /1 B 에서 1은 이름을 찾는 인덱스 값으로 쓰고, A와 B는 이름 옆에 리스트안에 리스트로 추가

- 그래서 리스트 길이를 따졌을 때 귀걸이를 받은 사람은 [이름, A, B] 구조로 리스트가 이뤄져있고, 받지 못한 사람은 [이름, A(혹은B)] 로 길이 차이가 발생한다.

- 길이가 2인 사람이 귀걸이를 받지 못했으므로 missing 리스트에 넣어주고, 시나리오가 끝나면 이 사람들을 형식에 맞게 출력해준다. 

missing = []
while True:
  student = int(input())
  stu_list = []
  if student == 0:
    break
  else:
    for _ in range(student):
      name = input()
      stu_list.append([name])

    for _ in range(student*2-1):
      num, AB = input().split()
      stu_list[int(num)-1].append(AB)

    for info in stu_list:
      if len(info) == 2:
        missing.append(info[0])

for idx, name in enumerate(missing):
  print(idx+1, name)