[BaekJoon] 2857번 : FBI (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 FBI입니다.

2857번: FBI

Solution

def find_fbi_agent(agent_list):
    result = "HE GOT AWAY!"
    fbi_agent_ids = []
    for agent_id, agent in enumerate(agent_list):
        if agent.find("FBI") != -1:
            fbi_agent_ids.append(str(agent_id + 1))
    
    if fbi_agent_ids != []:
        result = " ".join(fbi_agent_ids)
    return result

if __name__ == "__main__":
    agent_list = []
    for _ in range(5):
        agent = input()
        agent_list.append(agent)
    print(find_fbi_agent(agent_list))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07