[프로그래머스/파이썬] Lv.0 로그인 성공? (3')

Programmers 레벨 0_코딩테스트 입문  문제 풀이

https://school.programmers.co.kr/learn/challenges/beginner?order=recent&page=1 

코딩테스트 입문 | 프로그래머스 스쿨

 

풀이 날짜 : 2023-09-28

소요 시간 : 3min

 

---

 

문제 설명

머쓱이는 프로그래머스에 로그인하려고 합니다. 머쓱이가 입력한 아이디와 패스워드가 담긴 배열 id_pw와 회원들의 정보가 담긴 2차원 배열 db가 주어질 때, 다음과 같이 로그인 성공, 실패에 따른 메시지를 return하도록 solution 함수를 완성해주세요.
아이디와 비밀번호가 모두 일치하는 회원정보가 있으면 "login"을 return합니다.
로그인이 실패했을 때 아이디가 일치하는 회원이 없다면 “fail”를, 아이디는 일치하지만 비밀번호가 일치하는 회원이 없다면 “wrong pw”를 return 합니다.

 

 

 

나의 풀이

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

 

 

 

다른 풀이

def solution(id_pw, db):
    if db_pw := dict(db).get(id_pw[0]):
        return "login" if db_pw == id_pw[1] else "wrong pw"
    return "fail"

python 3.8 에서 := 연산자가 새로 나왔다.