[Python] 로그인 성공? - 프로그래머스

 

 

나의 문제 풀이

def solution(id_pw, db):
    answer = ''
    for i in db:
        if i[0] == id_pw[0] and i[1] == id_pw[1]:
            return 'login'
        if i[0] == id_pw[0] and i[1] != id_pw[1]:
            answer = 'wrong pw'
    if answer == '':
        return 'fail'
    return answer

 

여기 문제에서 의도하는 바는 bd 안에서 id_pw 하나라도 아이디 비밀번호가 같은 것이 있으면 login을 리턴하는 것이고 아이디라도 같은 것이 있으면(아이디 비밀번호 둘 다 같은 것이 없을 경우) wrong pw를 리턴하는 것입니다. 그리고 아무것도 해당되지 않았을 경우에는 fail를 리턴하는 것이죠

 

그래서 저는 코드를 이렇게 짰습니다.

먼저 for문을 돌리면서 아이디, 비밀번호 둘다 같은 것이 있으면 login를 리턴하고 한번이라도 아이디만 같을 경우에는 answer을 worng pw로 변경시킨 채로 루프를 도리는 것입니다. 그러나 이 둘 중 아무것도 해당되지 않으면 answer은 ''이 될 것입니다. 

 

그래서 answer이 ''일 경우 fail를 리턴 했고 answer이 worng pw이면 answer이 ''이 아니기 때문에 다음줄인 return answer을 통해 worng pw를 리턴하게 됩니다.