로그인 성공?

https://school.programmers.co.kr/learn/courses/30/lessons/120883

프로그래머스

---

문제 설명
머쓱이는 프로그래머스에 로그인하려고 합니다. 머쓱이가 입력한 아이디와 패스워드가 담긴 배열 id_pw와 회원들의 정보가 담긴 2차원 배열 db가 주어질 때, 다음과 같이 로그인 성공, 실패에 따른 메시지를 return하도록 solution 함수를 완성해주세요.

아이디와 비밀번호가 모두 일치하는 회원정보가 있으면 "login"을 return합니다.
로그인이 실패했을 때 아이디가 일치하는 회원이 없다면 “fail”를, 아이디는 일치하지만 비밀번호가 일치하는 회원이 없다면 “wrong pw”를 return 합니다.

---

제한사항

• 회원들의 아이디는 문자열입니다.
• 회원들의 아이디는 알파벳 소문자와 숫자로만 이루어져 있습니다.
• 회원들의 패스워드는 숫자로 구성된 문자열입니다.
• 회원들의 비밀번호는 같을 수 있지만 아이디는 같을 수 없습니다.
• id_pw의 길이는 2입니다.
• id_pw와 db의 원소는 [아이디, 패스워드] 형태입니다.
• 1 ≤ 아이디의 길이 ≤ 15
• 1 ≤ 비밀번호의 길이 ≤ 6
• 1 ≤ db의 길이 ≤ 10
• db의 원소의 길이는 2입니다.

---

입출력 예

id_pw	db	result
["meosseugi", "1234"]	[["rardss", "123"], ["yyoom", "1234"], ["meosseugi", "1234"]]	"login"
["programmer01", "15789"]	[["programmer02", "111111"], ["programmer00", "134"], ["programmer01", "1145"]]	"wrong pw"
["rabbit04", "98761"]	[["jaja11", "98761"], ["krong0313", "29440"], ["rabbit00", "111333"]]	"fail"

---

입출력 예 설명
입출력 예 #1

• db에 같은 정보의 계정이 있으므로 "login"을 return합니다.

입출력 예 #2

• db에 아이디는 같지만 패스워드가 다른 계정이 있으므로 "wrong pw"를 return합니다.

입출력 예 #3

• db에 아이디가 맞는 계정이 없으므로 "fail"을 return합니다.

---

id, pw가 주어진 db 내에서 일치하는지 확인하는 문제입니다.

 

굳이 db가 리스트일 필요는 없겠죠. 제 생각에는 db가 딕셔너리라면 좀 더 편하게 문제를 풀 수 있을 것 같습니다.

파이썬으로 문제를 풀어보겠습니다.

def solution(id_pw, db):
    # db를 딕셔너리 형태로 변환합니다.
    db_dict = dict(db)
    # id_pw를 db에서 확인합니다.
    uid, pw = id_pw
    
    # 아이디가 존재하는지 확인합니다.
    if uid not in db_dict.keys(): # 아이디가 존재하지 않는다면
        return "fail"
    else: # 아이디가 존재한다면
        if pw == db_dict[uid]: # 비밀번호가 일치한다면
            return "login"
        else: # 비밀번호가 일치하지 않는다면
            return "wrong pw"

db_dict이라는 딕셔너리를 만들어, key로 비교해서 아이디를 비교합니다.

딕셔너리로 푸는 이유는 결국 아이디는 유일한 하나뿐이고, 아이디와 비밀번호가 연결되어 있기 때문입니다.

만약 아이디가 존재하지 않는다면 "fail", 존재하더라도 비밀번호가 틀리면 "wrong pw", 비밀번호가 맞다면 "login"을 출력합니다.

 

딕셔너리로 키 - 값 구조를 구현하면 편하게 풀 수 있는 문제입니다.