[Python] 백준 2857번_FBI

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

2857번: FBI

 

• 5명의 요원 중 FBI 요원을 찾기
• FBI요원은 요원의 첩보원명에 FBI가 들어있음

• 입력
  • 5개 줄에 요원의 첩보원명이 주어짐
  • 첩보원명은 알파벳 대문자, 숫자 0~9, 대시 (-)로만 이루어져 있으며, 최대 10글자
• 출력
  • 첫째 줄에 FBI 요원을 출력
  • 이때, 해당하는 요원이 몇 번째 입력인지를 공백으로 구분하여 출력해야 하며, 오름차순으로 출력해야 함
  • 만약 FBI 요원이 없다면 "HE GOT AWAY!"를 출력
• 시간 제한: 0.25초
• 메모리 제한: 128 MB

 

"""
입출력 예시)

(예제 입력 1) 
N-FBI1
9A-USKOK
I-NTERPOL
G-MI6
RF-KGB1
        -> 1

(예제 입력 2) 
N321-CIA
F3-B12I
F-BI-12
OVO-JE-CIA
KRIJUMCAR1
            -> HE GOT AWAY!

(예제 입력 3) 
47-FBI
BOND-007
RF-FBI18
MARICA-13
13A-FBILL
        -> 1 3 5
"""

 

 

## 의사코드 ##

# 입력 받은 문자열에 'FBI'가 들어있으면 (index+1)을 리스트에 저장

# 저장한 리스트가 비어있으면 'HE GOT AWAY!' 출력
# 아니라면 오름차순으로 출력 
#     -> *리스트

 

 

 

통과한 코드

 

fbi = []
for i in range(5):
    name = input()
    if 'FBI' in name:
        fbi.append((i+1))
if len(fbi) == 0:
    print('HE GOT AWAY!')
else:
    print(*fbi)