[Python] 백준 2857번 FBI

문제

5명의 요원 중 FBI 요원을 찾는 프로그램을 작성하시오.

FBI요원은 요원의 첩보원명에 FBI가 들어있다.

 

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

2857번: FBI

입력

5개 줄에 요원의 첩보원명이 주어진다.

첩보원명은 알파벳 대문자, 숫자 0~9, 대시 (-)로만 이루어져 있으며, 최대 10글자이다.

출력

첫째 줄에 FBI 요원을 출력한다.

이때, 해당하는 요원이 몇 번째 입력인지를 공백으로 구분하여 출력해야 하며, 오름차순으로 출력해야 한다.

만약 FBI 요원이 없다면 "HE GOT AWAY!"를 출력한다.

예제 입력 1

N-FBI1
9A-USKOK
I-NTERPOL
G-MI6
RF-KGB1

예제 출력 1

1

예제 입력 2

N321-CIA
F3-B12I
F-BI-12
OVO-JE-CIA
KRIJUMCAR1

예제 출력 2

HE GOT AWAY!

문제 풀이

L = [input() for i in range(5)]

cnt = 0				# 요원이 없는지 판단하기 위한 변수
for i in range(5):
    if 'FBI' in L[i]:		# 조건문으로 'FBI'가 포함되어있는지 확인
        print(i+1, end=' ')	# 인덱스 +1로 몇번째 입력인지 출력, 요구 사항에 맞추어 공백으로 구분
        cnt += 1		# 0이 아닌 임의의 숫자로 지정해도 무방

if cnt == 0:
    print('HE GOT AWAY!')

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.