백준] 2857 - FBI(COCI 2010/2011)

시간 제한 : 1초

메모리 제한 : 128MB

입력

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

출력

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

소스코드

#include <iostream>
#include <string>
#include <vector>
using namespace std;
int main(void)
{
	int n = 5;
	vector<int> arr;
	
	for (int i = 1; i <= n; i++)
	{
		string str;
		cin >> str;
		if (str.find("FBI", 0) != string::npos)
			arr.push_back(i);
	}

	if (arr.empty())
		cout << "HE GOT AWAY!";
	else
		for (int i = 0; i < arr.size(); i++)
			cout << arr[i] << ' ';
}

Tip

문자열을 처리하는 문제로 find()와 string::npos 검출만 이용해도 풀 수 있다.