[programmers - python] Lv.0: 로그인 성공?

Interrobang 2022. 11. 29. 15:55

문제 링크

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 풀이

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

*key point: if문과 for문을 적절히 활용하여 주어진 조건에 따라 적합한 return 값을 가지도록 한다.