[Python] 백준 2857번: FBI

코드

n_list = []
for i in range(1, 6):
    n = input()
    if 'FBI' in n:
        n_list.append(i)
if n_list:
    print(*n_list)
else:
    print('HE GOT AWAY!')

설명

빈 리스트를 하나 생성한다.

반복문을 통해 입력을 받는다.

그 입력값 중에 FBI가 있으면 n_list에 번호를 넣는다.

번호를 넣어야하기 때문에 반복문의 범위는 (1, 6)이 된다.

만약 n_list에 값에 있다면 n_list를 출력한다.

리스트 앞에 *을 붙이고 출력을 하면 대괄호가 사라지고 내부에 있는 요소들만 출력하게 된다.

n_list안에 값이 없다면 'HE GOT AWAY!'를 출력한다.