2857번: FBI

문제 출처 : https://www.acmicpc.net/problem/2857

2857번: FBI

언어 : Kotlin

 

문제 설명 :

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

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

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

 

제한 사항 :

• 시간 제한 : 0.25초 (추가 시간 없음)
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
N-FBI1 9A-USKOK I-NTERPOL G-MI6 RF-KGB1	1
N321-CIA F3-B12I F-BI-12 OVO-JE-CIA KRIJUMCAR1	HE GOT AWAY!
47-FBI BOND-007 RF-FBI18 MARICA-13 13A-FBILL	1 3 5

 

풀이 : 

import java.io.BufferedWriter
import java.io.OutputStreamWriter

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val arr = arrayListOf<String>()
    repeat(5) { arr += readLine() }

    var idxList = intArrayOf()
    arr.forEachIndexed { index, s -> if (s.contains("FBI")) idxList += index + 1 }

    bw.write(if (idxList.isNotEmpty()) idxList.joinToString(" ") else "HE GOT AWAY!")
    bw.flush()
    bw.close()
}