본문 바로가기

Algorithm & SQL/Programmers

[Programmers] [Python] 서울에서 김서방 찾기

서울에서 김서방 찾기


문제 설명


String형 배열 seoul의 element중 Kim의 위치 x를 찾아, 김서방은 x에 있다는 String을 반환하는 함수, solution을 완성하세요. seoul에 Kim은 오직 한 번만 나타나며 잘못된 값이 입력되는 경우는 없습니다.

제한 사항


  • seoul은 길이 1 이상, 1000 이하인 배열입니다.
  • seoul의 원소는 길이 1 이상, 20 이하인 문자열입니다.
  • 'Kim'은 반드시 seoul 안에 포함되어 있습니다.

입출력 예


seoul return
["Jane", "Kim"] "김서방은 1에 있다"

###제출코드


def solution(seoul):
   for idx, name in enumerate(seoul):
       if name == 'Kim':
           return "김서방은 {}에 있다".format(idx)

코드설명


이 문제도 매우 쉬웠다.

for문을 돌면서 enumerate를 이용해 idxname을 뽑아내어 name값이 Kim인 경우,

해당 idx를 반환하도록 하였다.

다른 사람의 풀이


def solution(seoul):
   return "김서방은 {}에 있다".format(seoul.index('kim'))

아... index()의 존재를 까먹었었다..

seoul string에서 index()를 통해 바로 kim을 찾아냈다.