알고리즘 풀이/프로그래머스

[Level1] 서울에서 김서방 찾기

mhko411 2021. 1. 21. 14:40
728x90

문제

여러 개의 문자열을 갖는 리스트가 입력되었을 때 "Kim"을 찾아본다.

"Kim"이 존재하는 인덱스를 찾아 "김서방은 (인덱스)에 있다."를 출력해보자.

 

입력

["Jane", "Kim"]

 

출력

"김서방은 1에 있다"


1. "Kim"의 인덱스를 저장할 idx를 선언하고 0으로 초기화한다.

2. seoul 리스트에서 원소를 뽑아서 "Kim"을 찾는 탐색을 진행하고 idx을 1씩 증가시킨다.

3. "Kim"을 찾았을 때 break를 하여 탐색을 종료하고

4. answer에 "김서방은 {}에 있다".format(idx)를 대입한다.

 

def solution(seoul):
    answer = ''
    idx=0
    for name in seoul:
        if name=='Kim':
            break
        idx+=1
    answer="김서방은 {}에 있다".format(idx)
    return answer