[프로그래머스Lv0] 로그인 성공? - 파이썬

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

프로그래머스

<문제접근법>

3가지의 종류가 있다.

아이디와 패스워드가 동일해서 로그인을 하거나 , 패스워드만 틀리거나 , 아이디/비번이 모두 틀린경우가 존재한다.

먼저 아이디와 패스워드가 동일한 경우 "login"을 출력하고 아닌경우는 나눈다

아닌 경우중에서 db에서 0번째 인덱스만 같으면 아이디만 같은 것으로 "wrong pw"로 출력하고 

그것이 아닌 경우에는 "fail"로 출력을 한다.

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