[백준/BOJ/C++] 2857번 FBI

[문제]

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

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

 

[입력]

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

 

[출력]

첫째 줄에 FBI 요원을 출력한다. 이때, 해당하는 요원이 몇 번째 입력인지를 공백으로 구분하여 출력해야 하며, 오름차순으로 출력해야 한다. 만약 FBI 요원이 없다면 "HE GOT AWAY!"를 출력한다.

 

[Source Code]

#include <stdio.h>
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;

int main() {
	int arr[] = {0,0,0,0,0};
	int count = 0;

	for(int i = 0; i < 5; i++){
		string input;
		cin >> input;

		for(int j = 0; j < input.length()-2; j++){
			if(input[j] == 'F' && input[j+1] == 'B' && input[j+2] == 'I'){
				arr[i] = i+1;
				count = 1;
			}
		}
	}

	for(int i = 0; i < 5; i++){
		if(count == 0){
			cout << "HE GOT AWAY!";
			break;
		}
		else if(arr[i] != 0)
			cout << arr[i] << " ";
	}
}

[결과 화면]